diff --git a/run_model.py b/run_model.py index 824a183..434acea 100644 --- a/run_model.py +++ b/run_model.py @@ -14,18 +14,20 @@ with open(args.config_path, "r") as cfgfile: config = yaml.load(cfgfile) -# TRAIN + TEST +# MANUAL: TRAIN + TEST # model = SconeClassifier(args.config_path) # _, history = model.train() # pd.DataFrame(history.history).to_csv(os.path.join(OUTPUT_PATH, "training_history.csv")) # test_acc = model.test() -# TRAIN + PREDICT +# MANUAL: TRAIN + PREDICT # trained_model, history = model.train() # test_set, test_ids = model.get_test_set() # trained_model.predict(test_set, test_ids) +# AUTOMATIC: RUNS TRAIN/TEST/PREDICT AS SPECIFIED BY CONFIG preds_dict, history = SconeClassifier(config).run() + print("######## CLASSIFICATION REPORT ########") print("classification finished in {:.2f}min".format((time.time() - start) / 60)) print("last training accuracy value: {}".format(history.history["accuracy"][-1]))