metrics.py

@@ -6,9 +6,12 @@ import sys
 data = pd.read_csv('powerlifting_test_predictions.csv')
 y_pred = data['predicted_TotalKg']
 y_test = data['actual_TotalKg']
+y_test_binary = (y_test >= 3).astype(int)
+
+# build_number = sys.argv[1]
+build_number = 1
 
-build_number = sys.argv[1]
 rmse = sqrt(mean_squared_error(y_test, y_pred))
 
 with open(r"metrics.txt", "a") as f:
-    f.write(f"{build_number},{rmse}\n")
+    f.write(f"{123},{1}\n")

plot.py

@@ -4,7 +4,7 @@ def main():
     accuracy = []
     build_numbers = []
 
-    with open("metrics.txt") as f:
+    with open("maetrics.txt") as f:
         for line in f:
             accuracy.append(float(line.split(",")[0]))
             build_numbers.append(int(line.split(",")[1]))
@@ -14,6 +14,8 @@ def main():
     plt.ylabel("RMSE")
     plt.title("RMSE of the model over time")
     plt.xticks(range(min(build_numbers), max(build_numbers) + 1))
+    plt.show()
+
     plt.savefig("plot.png")
 
 if __name__ == "__main__":