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

update

Browse Source
This commit is contained in:
Newnius
2020-05-01 18:47:41 +08:00
parent 011bdf146c
commit 91c55959ff
2 changed files with 30 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
serve.py
View File

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