Update main.py

CMATER-JUCS · Feb 23, 2022 · ca359eb · ca359eb
1 parent fd5d0dc
commit ca359eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -26,5 +26,6 @@
 #n is the number of folds
 n = len(files_for_train_x)
 for i in range(0,n):
-
-     k_fold_separate(files_for_train_x[i] , files_for_train_y[i] , files_for_validation_x[i] ,files_for_validation_y[i] , "InceptionV3" , "MobileNetV2" ,"InceptionResNetV2" ,i+1 , NUM_EPOCHS = args.num_epochs , train_batch=args.batch_size , validation_batch = args.batch_size, lr=args.lr) 
+    k_fold_separate(files_for_train_x[i] , files_for_train_y[i] , files_for_validation_x[i] ,files_for_validation_y[i] ,
+                    "InceptionV3" , "MobileNetV2" ,"InceptionResNetV2" ,i+1 ,
+                    NUM_EPOCHS = args.num_epochs , train_batch=args.batch_size , validation_batch = args.batch_size, lr=args.lr)