From 906f3388a52deeaa8e209c31e8969abeb258e96b Mon Sep 17 00:00:00 2001 From: Newnius Date: Wed, 29 Apr 2020 19:00:06 +0800 Subject: [PATCH] add files --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 51505a6..173145d 100644 --- a/main.py +++ b/main.py @@ -166,7 +166,7 @@ def main(config): train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data() model = train(config, train_X, train_Y, valid_X, valid_Y) - if config.do_predict: + test_X, test_Y = data_gainer.get_test_data(return_label_data=True) pred_result = predict(config, test_X, model) draw(config, data_gainer, pred_result)