인프런 커뮤니티 질문&답변
작성한 질문수
안녕하세요
작성
·
237
0
안녕하세요. 정신없이 듣다보니 중요한걸 놓친거 같아 질문드립니다.
현재 선형회귀 비슷한 모델을 데이터를 구해서 연습삼아 만들어본뒤 실제 데이터 몇개를 넣어서 얼마나 잘되는지…결과값을 받아 보려고 하고있는데
이 몇 개의 테스트 데이터 (타겟값 제외)를 아래 어디다가 넣어야할지를 모르겠습니다 (csv파일)
Predict로 결과값을 받아보려고 할 때, X_test와 y_test에 넣으면 되는걸까요?
X_train
X_test
y_trian
y_test
감사합니다.
답변 1
1
권 철민
지식공유자
안녕하십니까,
X_train, y_train으로 모델을 학습한 뒤에 해당 모델 regressor, 가령 Ridge로 학습된 ridge_regressor 객체가 있으면 해당 객체의 predict( ) 메소드에 인자로 테스트 데이터 값을 입력하면 됩니다.
y_pred = ridge_regressor.predict(X_test) # 여기서 X_test는 target값 제외 테스트 데이터 입니다.
y_pred는 반환된 예측 결과이므로 테스트 데이터와의 실제 값과 비교해 보시면 됩니다.
감사합니다.