From abe9929c47c4cb91bad9df6c953c29f2baea4306 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 12:55:42 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 19 +++++++++---------- train.py | 4 ++-- 2 files changed, 11 insertions(+), 12 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 591d687..60a8d09 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -47,7 +47,7 @@ - + @@ -56,7 +56,7 @@ - + @@ -91,8 +91,8 @@ - - + + @@ -226,7 +226,7 @@ - + @@ -267,16 +267,15 @@ - - @@ -398,8 +397,8 @@ - - + + diff --git a/train.py b/train.py index 0974f1b..82f2f9f 100644 --- a/train.py +++ b/train.py @@ -90,7 +90,7 @@ def experiment(repeats, series, seed): raw_values = series.values diff_values = difference(raw_values, 1) # transform data to be supervised learning - lag2 = 10 + lag2 = 4 supervised = timeseries_to_supervised(diff_values, lag2) supervised_values = supervised.values # split data into train and test-sets @@ -103,7 +103,7 @@ def experiment(repeats, series, seed): error_scores = list() for r in range(repeats): # fit the model - batch_size = 4 + batch_size = 32 t1 = train.shape[0] % batch_size t2 = test.shape[0] % batch_size