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:13:06 +08:00
parent 7cd43511ac
commit 557ee80026
2 changed files with 31 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
serve.py
View File

@@ -142,28 +142,35 @@ def draw_yqy(config, origin_data, predict_norm_data, mean_yqy, std_yqy):
assert label_norm_data.shape[0] == predict_norm_data.shape[
0], "The element number in origin and predicted data is different"
print("dasdsadsadas")
# label_norm_data=label_norm_data[:,1]
label_name = 'high'
label_names = ['pre', 'main', 'post']
label_column_num = 1
loss = \
np.mean((label_norm_data[config.predict_day:][:, 1] - predict_norm_data[:-config.predict_day][0:]) ** 2,
axis=0)[1]
print("The mean squared error of stock {} is ".format(label_name), loss)
loss1 = \
np.mean((label_norm_data[config.predict_day:][:, 5] - predict_norm_data[:-config.predict_day][0:]) ** 2,
axis=0)[5]
loss2 = \
np.mean((label_norm_data[config.predict_day:][:, 6] - predict_norm_data[:-config.predict_day][0:]) ** 2,
axis=0)[6]
loss3 = \
np.mean((label_norm_data[config.predict_day:][:, 7] - predict_norm_data[:-config.predict_day][0:]) ** 2,
axis=0)[7]
print("The mean squared error of stock {} is ".format(label_names[0]), loss1)
print("The mean squared error of stock {} is ".format(label_names[1]), loss2)
print("The mean squared error of stock {} is ".format(label_names[2]), loss3)
# label_X = range(origin_data.data_num - origin_data.train_num - origin_data.start_num_in_test)
# predict_X = [x + config.predict_day for x in label_X]
label_data = label_norm_data[:, 1] * std_yqy[1] + mean_yqy[1]
label_datas = label_norm_data * std_yqy + mean_yqy
predict_data = predict_norm_data * std_yqy[1] + mean_yqy[1]
predict_datas = predict_norm_data * std_yqy + mean_yqy
print(label_data)
print(predict_data)
print(label_datas)
print(predict_datas)
print(label_data[-1])
print(predict_data[-1])
print(label_datas[-1])
print(predict_datas[-1])
PORT_NUMBER = 8080