update

Author: bmalbakri

__pycache__/Data_processing.cpython-310.pyc

@@ -10,7 +10,7 @@ def __init__(self):
         define parameters for training the neural network
         """
         self.paramters = {
-            'Data Size': int(10),
+            'Data Size': int(1e3),
             'save_data': False,
             'Frequency_lb': -3,
             'Frequency_ub': 8,

__pycache__/NN.cpython-310.pyc

@@ -8,6 +8,7 @@
 import sklearn
 import pandas as pd
 from sklearn.metrics import accuracy_score
+import os
 warnings.filterwarnings("ignore")
 
 
@@ -45,7 +46,10 @@ def predict(self):
         Confidence = Confidence.cpu().detach().numpy()
         degree_prediction = degree_prediction.cpu().detach().numpy()
         binary_prediction = binary_prediction.cpu().detach().numpy()
+        import os
 
+        folder_name = 'predictions'
+        os.makedirs(folder_name, exist_ok=True)
         np.savetxt('predictions/Pred_Binary_prob', binary_prediction)
         np.savetxt('predictions/Pred_Binary', np.round(binary_prediction))