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:51:08 +08:00
parent 91c55959ff
commit f119fd345e
2 changed files with 31 additions and 46 deletions

View File

@@ -148,8 +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:][:, 5][:,np.newaxis] - predict_norm_data[:-config.predict_day][0:]) ** 2,
axis=0)
np.mean((label_norm_data[config.predict_day:][:-3] - 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)