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

강사님 안녕하세요.

강사님께서 말씀해주신 랜포 + train, test = train.align(test, join='left', axis=1) 하고나서

에러메시지가 뜨길래 추가적인 fillna(0)를 해주고, 아래와 같이 작성했는데요

문제에서 제시한 예시와 제 답안의 수치가 너무 차이가 많이납니다.

어떤부분이 문제일까요? ㅜ_ㅠ 확인 부탁드립니다.

isnull 에 무조건 0채우라고해서 fillna(0)한거밖에 차이가 없습니다.

import pandas as pd

train = pd.read_csv("data/customer_train.csv")

test = pd.read_csv("data/customer_test.csv")

# print(train.info())

# print(test.info())

train = train.fillna(0)

test = test.fillna(0)

# df.isnull().sum()

target = train.pop('총구매액')

train = pd.get_dummies(train)

test = pd.get_dummies(test)

train, test = train.align(test, join='left', axis=1)

train = train.fillna(0)

test = test.fillna(0)

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(test)

submit = pd.DataFrame({'pred':pred})

submit.to_csv('result.csv', index=False)

ans = pd.read_csv('result.csv')

print(ans)