このコードを実行すると、エラー_KeyError: '[ 1351 1352 1353 ... 13500 13501 13502] not in index'
_が発生する理由がわかりません。
_cv = KFold(n_splits=10)
for train_index, test_index in cv.split(X):
f_train_X, f_valid_X = X[train_index], X[test_index]
f_train_y, f_valid_y = y[train_index], y[test_index]
_
X
(a Pandas dataframe)を使用して、I cv.split(X)
を分割します。
_X.shape
y.shape
Out: (13503, 17)
Out: (13503,)
_
問題は、X[train_index]
_を使用してX
をインデックス化しようとしている方法です。_.loc
_または_.iloc
_ pandas
データフレームがあるため。
_cv = KFold(n_splits=10)
for train_index, test_index in cv.split(X):
f_train_X, f_valid_X = X.iloc[train_index], X.iloc[test_index]
f_train_y, f_valid_y = y.iloc[train_index], y.iloc[test_index]
_
iloc
を使用した例_import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))
df[[1,2]]
#KeyError: '[1 2] not in index'
df.iloc[[1,2]]
# A B C D
#1 25 97 78 74
#2 6 84 16 21
_
_df = df.values
#now this should work fine
df[[1,2]]
#array([[25, 97, 78, 74],
# [ 6, 84, 16, 21]])
_