From 91c55959ff9d552cd72edc9fa27f427b4957f551 Mon Sep 17 00:00:00 2001 From: Newnius Date: Fri, 1 May 2020 18:47:41 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 54 +++++++++++++++++++++++---------------------- serve.py | 3 ++- 2 files changed, 30 insertions(+), 27 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 30c7195..7cc7460 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -3,6 +3,7 @@ + @@ -48,7 +49,7 @@ - + @@ -57,18 +58,18 @@ - + - + - - + + @@ -85,14 +86,14 @@ - + - - + + - + @@ -168,11 +169,11 @@ @@ -215,7 +216,7 @@ - + @@ -256,15 +257,16 @@ - + @@ -353,16 +355,6 @@ - - - - - - - - - - @@ -375,11 +367,21 @@ - - + + - + + + + + + + + + + + diff --git a/serve.py b/serve.py index b1ee2ac..f733f65 100644 --- a/serve.py +++ b/serve.py @@ -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)