강의

멘토링

커뮤니티

인프런 커뮤니티 질문&답변

co choi님의 프로필 이미지
co choi

작성한 질문수

[개정판] 파이썬 머신러닝 완벽 가이드

선생님 kfold 질문있습니다.

작성

·

249

0

강의 너무 잘듣고 있습니다 항상 감사드립니다. 

kfold 로 데이터나눌때 데이터프레임으로 변환하고 데이터를 넣으니까 이런오류가 뜨는데 어떻게 해야하나요??

kfold뒤 매개변수에 데이터프레임은 안되고 numpy 형태만 가능한가요??

자세한 설명 부탁드립니다 ㅠㅠ 

답변 1

1

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까, 

아마도 iloc등으로 찾지 않아서 오류가 나오는 것 같습니다. 

아래와 같이 수정해 보시지요. 

iris_data = load_iris()

iris_df = pd.DataFrame(iris_data.data, columns=iris_data.feature_names)

label_df = pd.DataFrame(iris_data.target, columns=['target'])

for train_index, test_index  in kfold.split(iris_df):

    #print(train_index, test_index)

    X_train, X_test = iris_df.iloc[train_index], iris_df.iloc[test_index]

    y_train, y_test = label_df.iloc[train_index], label_df.iloc[test_index]

    print('#### features:\n', X_train.iloc[0], '\n#### target:', y_train.iloc[0])

co choi님의 프로필 이미지
co choi

작성한 질문수

질문하기