YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 00:16:44 +00:00
Code Issues Releases Wiki Activity

add files

Browse Source
This commit is contained in:
Newnius
2020-04-29 19:00:06 +08:00
parent 8fbce50cf8
commit 906f3388a5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -166,7 +166,7 @@ def main(config):
train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data() 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) 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) test_X, test_Y = data_gainer.get_test_data(return_label_data=True)
pred_result = predict(config, test_X, model) pred_result = predict(config, test_X, model)
draw(config, data_gainer, pred_result) draw(config, data_gainer, pred_result)