Parcourir la source

Add of validation split info

Jérôme BUISINE il y a 4 ans
Parent
commit
3369b67e76
1 fichiers modifiés avec 3 ajouts et 1 suppressions
  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:]