1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 00:16:44 +00:00
This commit is contained in:
2020-05-01 18:53:27 +08:00
parent f119fd345e
commit 25236a77d4
2 changed files with 2 additions and 2 deletions

View File

@@ -148,7 +148,7 @@ def draw_yqy(config2, origin_data, predict_norm_data, mean_yqy, std_yqy):
label_column_num = 3
loss = \
np.mean((label_norm_data[config.predict_day:][:-3] - predict_norm_data[:-config.predict_day]) ** 2, axis=0)
np.mean((label_norm_data[config.predict_day:, 5:8] - predict_norm_data[:-config.predict_day]) ** 2, axis=0)
print("The mean squared error of stock {} is ".format(label_name), loss)
# label_X = range(origin_data.data_num - origin_data.train_num - origin_data.start_num_in_test)