Преглед на файлове

Add of validation split info

Jérôme BUISINE преди 5 години
родител
ревизия
3369b67e76
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  1. 3 1
      train_model.py

+ 3 - 1
train_model.py

@@ -111,8 +111,10 @@ def main():
     final_df_val_size = len(final_df_val.index)
 
     validation_split = final_df_val_size / (final_df_train_size + final_df_val_size)
+    print("----------------------------------------------------------")
     print("Validation size is based of `.val` content")
-    print("Validation split is now set at", )
+    print("Validation split is now set at", validation_split)
+    print("----------------------------------------------------------")
 
     # use of the whole data set for training
     x_dataset_train = final_df_train.ix[:,1:]