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

update

Browse Source
This commit is contained in:
Newnius
2020-05-02 01:49:15 +08:00
parent e6bc92507c
commit daf04a03c1
2 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
serve.py
View File

@@ -21,7 +21,7 @@ class Config:
predict_day = 1
input_size = len(feature_columns)
input_size = len(feature_columns)-1
output_size = len(label_columns)
hidden_size = 128
@@ -182,7 +182,7 @@ def train_models():
if train_X.shape[0] < 200:
config.batch_size = 16
print(train_X[:, :1], valid_X[:, :1], train_Y, valid_Y)
print(train_X[:1, ], valid_X[:1, :], train_Y, valid_Y)
train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y)
lock.release()