Update app.py

app.py

@@ -12,6 +12,9 @@ warnings.filterwarnings("ignore")
 X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=5000, random_state=42)
 data_split = StratifiedShuffleSplit(n_splits=1, test_size=0.9, random_state=0) # split data one time into two parts with ratio 10%/90%
 for i, (train_index, test_index) in enumerate(data_split.split(X_train, y_train)):
+   print(f"Fold {i}:")
+   print(f" Train: index={train_index}, size = {len(train_index)}")
+   print(f" Remaining: index={test_index}, size = {len(test_index)}")
 
 small_X_train = X_train[train_index]
 small_y_train = y_train[train_index]