update

christopher192 · Nov 14, 2023 · f866e73 · f866e73
1 parent 9f5aec8
commit f866e73
Show file tree

Hide file tree

Showing 2 changed files with 613 additions and 64 deletions.
diff --git a/helper_function.py b/helper_function.py
@@ -163,4 +163,10 @@ def plot_prediction(train_data, train_label, test_data, test_label, prediction =
     if predictions is not None:
         plt.scatter(test_data, prediction, c = "r", s = 4, label = "Prediction")
 
-    plt.legend(prop = {"size" : 14})
+    plt.legend(prop = {"size" : 14})
+
+def accuracy_function(y_true, y_pred):
+    correct = torch.eq(y_true, y_pred).sum().item()
+    acc = (correct / len(y_pred)) * 100
+
+    return acc