質問編集履歴
1
train_x,train_yを作る関数を記載しました
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -51,4 +51,23 @@
|
|
|
51
51
|
2.74E-04
|
|
52
52
|
3.02E-05
|
|
53
53
|
-2.32E-04
|
|
54
|
+
```
|
|
55
|
+
|
|
56
|
+
train_xを作る関数です。
|
|
57
|
+
```python
|
|
58
|
+
def load_data(cls, n_prev):
|
|
59
|
+
|
|
60
|
+
df = pd.DataFrame(list(daily_list.objects.filter().values()))
|
|
61
|
+
df = df.sort_values(['symbol', 'created_date_at'])
|
|
62
|
+
df = df.reset_index(drop=True)
|
|
63
|
+
trainX, trainY = [], []
|
|
64
|
+
|
|
65
|
+
for symbol, df_s in df.groupby('symbol'):
|
|
66
|
+
df_s2 = df_s.loc[:, 'signal0']
|
|
67
|
+
for i in range(len(df_s2) - n_prev):
|
|
68
|
+
trainX.append(df_s2.iloc[i:i + n_prev].values)
|
|
69
|
+
trainY.append(df_s2.iloc[i + n_prev])
|
|
70
|
+
atrainX = np.array(trainX)
|
|
71
|
+
atrainY = np.array(trainY)
|
|
72
|
+
return (atrainX, atrainY)
|
|
54
73
|
```
|