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

문상혁님의 프로필 이미지

작성한 질문수

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

평가지표 (이진/다중 분류, 회귀) 🆕 updated 2023.10

5회 작업형2 rmse 질문입니다

해결된 질문

24.06.11 16:01 작성

·

76

0

  • 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!

  • 질문과 관련된 영상 위치를 알려주면 더 빠르게 답변할 수 있어요

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

target = train.pop("price") 
train = pd.get_dummies(train) 
test = pd.get_dummies(test) 
 
from sklearn.model_selection import train_test_split 
X_tr, X_val, y_tr, 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)  

from sklearn.metrics import mean_squared_error 
mse = mean_squared_error(y_val, pred)  
rmse = mse **0.5
 print(rmse)  

이런식으로 rmse를 만들면 될까요??

답변 1

0

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

2024. 06. 11. 20:05

네 맞습니다. 잘 만드셨어요!