回答編集履歴
1
追加
answer
CHANGED
|
@@ -4,9 +4,9 @@
|
|
|
4
4
|
X_fit , X_val = X_train.iloc[train_index], X_train.iloc[val_index]
|
|
5
5
|
y_fit , y_val = y_train.iloc[train_index], y_train.iloc[val_index]
|
|
6
6
|
```
|
|
7
|
-
pandas.DataFrameな
|
|
7
|
+
pandas.DataFrameでindexにより要素にアクセスするには.ilocなどを使って上のようにする必要があります。
|
|
8
8
|
|
|
9
|
-
もしくはnp.arrayにす
|
|
9
|
+
もしくはnp.arrayにすれば以下のようにもできます。
|
|
10
10
|
```python
|
|
11
11
|
kf = KFold(n_splits=2)
|
|
12
12
|
for train_index, val_index in kf.split(X_train, y_train):
|