modif erreurs

script.sl

@@ -40,4 +40,4 @@ mkdir callbacks
 # export LOCAL_WORK_DIR=checkpoints
 
 # Start the calculation
-srun python3 urban_classif.py
\ No newline at end of file
+srun python3 urbanclass_fold.py
\ No newline at end of file

urbanclass_fold.py

@@ -130,12 +130,12 @@ def process_fold(fold_k, data, epochs=100, num_batch_size=32):
 
 
 if __name__ == "__main__":
-    
+    us8k_df = pd.read_pickle("us8k_df.pkl")
     history1 = []
     for i in range(10) : 
         FOLD = i+1
         
-        history = process_fold(FOLD_K, us8k_df, epochs=3)
+        history = process_fold(FOLD, us8k_df, epochs=100)
         history1.append(history)