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

세 가지 질문이 있는데요..

1) 이 문제 해결에 있어서 train_test_split은 하면 좋지만 굳이 안해도 되는 내용이 맞나요?

2) 아래와 같이 x_train, y_train 으로 학습을 하고 x_test에 대해서 예측을 하면 될까요?

3) 문제에서 보면 roc_auc_score로 평가한다 했는데, 이건 출제자가 평가하기 때문에 이 문제상에서는 단순 확인 및 학습을 위해 써본거지 답에는 필요 없었던 부분이 맞을까요?

model = RandomForestClassifier(random_state=2022)
model.fit(X_train, y_train)
pred = model.predict_proba(X_test)

submit = pd.DataFrame(
            {
                'cust_id':cust_id,
                'gender':pred[:,1]
            }
        )

from sklearn.ensemble import RandomForestClassifier

model = RandomForestClassifier(random_state=2022)
model.fit(X_tr, y_tr)
pred = model.predict_proba(X_val)

# 검증 데이터 셋으로 평가
from sklearn.metrics import roc_auc_score
roc_auc_score(y_val, pred[:,1])

# test 데이터 예측
pred = model.predict_proba(X_test)
pred

# 데이터 프레임 만들기
submit = pd.DataFrame(
            {
                'cust_id':cust_id,
                'gender':pred[:,1]
            }
        )