Commit c5103d2a authored by David Maxence's avatar David Maxence
Browse files

modif erreurs

parent 36622cab
......@@ -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
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment