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

rkgml5153님의 프로필 이미지

작성한 질문수

[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

7회 기출유형(작업형2) 🆕 updated 2024.4

7회 기출유형(작업형2) 오류 질문

해결된 질문

24.06.16 10:50 작성

·

75

0

import pandas as pd
train = pd.read_csv("mart_train.csv")
test = pd.read_csv("mart_test.csv")

#train.head()
#test.head()
#train.isnull().sum()
#test.isnull().sum()
#train.info()

target = train.pop('total')

#print(train.shape, test.shape)
train = pd.get_dummies(train)
test = pd.get_dummies(test)
#print(train.shape, test.shape)

from sklearn.model_selection import train_test_split
X_tr, y_tr, X_val, y_val = train_test_split(train, target, test_size=0.2, random_state=0)

from sklearn.ensemble import RandomForestRegressor
rf = RandomForestRegressor(random_state=0)
rf.fit(X_tr, y_tr)
pred = rf.predict(X_val)
  • 질문과 관련된 영상 위치를 알려주면 더 빠르게 답변할 수 있어요

  • 먼저 유사한 질문이 있었는지 검색해보세요

강의 보면서 따라했는데,

ValueError: Found input variables with inconsistent numbers of samples: [560, 140]

이렇게 에러가 뜨는데, 이유를 못찾겠습니다.

어떤게 문제인지 알 수 있을까요? 감사합니다.

답변 1

0

퇴근후딴짓님의 프로필 이미지
퇴근후딴짓
지식공유자

2024. 06. 16. 20:34

X_tr, y_tr, X_val, y_val

가 아니라

X_tr, X_val, y_tr, y_val

입니다.

암기 방법은

X, X, y, y를 먼저 적고

그리고 tr과 val 반복입니다.