1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-15 17:06:44 +00:00
This commit is contained in:
2020-05-02 18:07:41 +08:00
parent c36fa48b9f
commit af36795da6
2 changed files with 18 additions and 23 deletions

View File

@@ -135,19 +135,12 @@ def predict(job, seq):
return -1, False
# load dataset
data = {
'seq': seq,
'value': 0,
}
batch_size = int(models[job]['batch_size'])
df = read_csv('./data/' + job + '.csv', header=0, index_col=0, squeeze=True)
df = df.tail(batch_size * 2 - 1)
df.loc[df.shape[0]] = {
'seq': seq,
'value': 0,
}
df.loc[df.shape[0]] = [seq, 0]
print(df[0].shape)