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

안녕하세요

작업형 2관련 마지막으로 시험환경으로 연습해보고 있는데 갑자기 'killed' 오류 메세지가 나면서 모델링이 되지 않습니다. 기존 코랩과 비슷하게 했는데 어떤 차이가 있을까요??

import pandas as pd

X_test = pd.read_csv("data/X_test.csv")

X_train = pd.read_csv("data/X_train.csv")

y_train = pd.read_csv("data/y_train.csv")

# 사용자 코딩

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

pd.options.display.float_format = '{:.3f}'.format

X_train = X_train.drop('cust_id', axis = 1)

cust_id = X_test.pop('cust_id')

X_train = X_train.fillna(0)

X_test = X_test.fillna(0)

cols_n = ['총구매액', '최대구매액', '환불금액', '내점일수', '내점당구매건수', '주말방문비율', '구매주기']

cols_c = ['주구매상품', '주구매지점']

from sklearn.preprocessing import RobustScaler

scaler = RobustScaler()

X_train[cols_n] = scaler.fit_transform(X_train[cols_n])

X_test[cols_n] = scaler.transform(X_test[cols_n])

from sklearn.preprocessing import LabelEncoder

for col in cols_c:

le = LabelEncoder()

X_train[col] = le.fit_transform(X_train[col])

X_test[col] = le.transform(X_test[col])

from sklearn.ensemble import RandomForestClassifier

model = RandomForestClassifier(random_state = 38)

model.fit(X_train, y_train)

pred = model.predict_proba(X_test)