1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 08:26:43 +00:00
This commit is contained in:
2020-05-02 01:52:46 +08:00
parent daf04a03c1
commit 75cbdfda69
2 changed files with 20 additions and 21 deletions

View File

@@ -21,7 +21,7 @@ class Config:
predict_day = 1
input_size = len(feature_columns)-1
input_size = len(feature_columns) - 1
output_size = len(label_columns)
hidden_size = 128
@@ -128,7 +128,7 @@ class Data:
test_data_yqy = []
# test_data_yqy=test_data_yqy[1:21]
feature_data = (test_data_yqy - self.mean) / self.std
test_x = [feature_data][:, 0:1]
test_x = [feature_data]
print(test_x)
return np.array(test_x)
@@ -182,8 +182,8 @@ def train_models():
if train_X.shape[0] < 200:
config.batch_size = 16
print(train_X[:1, ], valid_X[:1, :], train_Y, valid_Y)
train(config, train_X[:, :1], train_Y, valid_X[:, :1], 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()