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

import pandas as pd

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

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

pd.set_option('display.max_columns', None)

# print(train.shape, test.shape)

# print(train.head())

# print(test.head())

# print(train.info())

# print(train.isnull().sum())

# print(test.isnull().sum())

train['환불금액'] = train['환불금액'].fillna(0)

test['환불금액'] = test['환불금액'].fillna(0)

cols = ['회원ID', '총구매액', '최대구매액','환불금액','방문일수','방문당구매건수','주말방문비율','구매주기']

# print(train.isnull().sum())

# print(test.isnull().sum())

target = train.pop('성별')

# 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, X_val, y_tr, y_val = train_test_split(train[cols],target, test_size = 0.2, random_state = 2022)

print(X_tr.shape, X_val.shape, y_tr.shape, y_val.shape)

from sklearn.ensemble import RandomForestClassifier

rf = RandomForestClassifier(random_state = 2022)

rf.fit(X_tr[cols], y_tr)

pred = rf.predict_proba(X_val[cols])

# print(pred)

from sklearn.metrics import roc_auc_score

print(roc_auc_score(y_val,pred[:,1]))

pred = rf.predict_proba(test[cols])

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

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

print(pd.read_csv('result.csv'))

그리고 코랩에서 연습할 때는 submit.to_csv('result.csv', index=False) 해당 코드를 작성하면, 파일 칸에 생기는 것을 확인했었는데, 여기서는 따로 생성이 안되고 그냥 제출하기만 하면 될까요?