YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-15 09:06:43 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius
2020-05-02 16:20:46 +08:00
parent 2dde7d48e7
commit f19b1b3ef8
2 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
train.py
View File

@@ -92,7 +92,11 @@ def experiment(repeats, series, seed):
supervised = timeseries_to_supervised(diff_values, lag2)
supervised_values = supervised.values
test_data_num = 32
batch_size = 32
if supervised_values.shape[0] < 100:
batch_size = 16
test_data_num = batch_size
# split data into train and test-sets
train, test = supervised_values[0:-test_data_num], supervised_values[-test_data_num:]
# transform the scale of the data
@@ -103,7 +107,6 @@ def experiment(repeats, series, seed):
error_scores = list()
for r in range(repeats):
# fit the model
batch_size = 32
t1 = train.shape[0] % batch_size
t2 = test.shape[0] % batch_size