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

Update of auc train score into backup model

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

+ 1 - 1
train_model.py

@@ -152,7 +152,7 @@ def main():
         os.makedirs(model_backup_folder)
 
     # add of callback models
-    filepath = os.path.join(cfg.backup_model_folder, p_output, p_output + "-{val_auc:02f}__{epoch:02d}.hdf5")
+    filepath = os.path.join(cfg.backup_model_folder, p_output, p_output + "-{auc:02f}-{val_auc:02f}__{epoch:02d}.hdf5")
     checkpoint = ModelCheckpoint(filepath, monitor='val_auc', verbose=1, save_best_only=True, mode='max')
     callbacks_list = [checkpoint]