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

안녕하세요 8회 작업형 2번을 간단한 방법을 풀다가 오류가 발생했는데 도통 뭐 잘못작성한지 모르겠습니다..

확인 부탁드리겠습니다!

import pandas as pd
train = pd.read_csv("https://raw.githubusercontent.com/lovedlim/inf/refs/heads/main/p4/8_2/churn_train.csv")
test = pd.read_csv("https://raw.githubusercontent.com/lovedlim/inf/refs/heads/main/p4/8_2/churn_test.csv")

# print(train.shape, test.shape)
# train.head()
# train.isnull().sum()

# train = train.drop('CustomerID', axis = 1)
# tesT = test.drop('CustomerID', axis = 1)
target = train.pop('TotalCharges')

# cols = train.select_dtypes(include = 'O').columns
# # print(cols)
# from sklearn.preprocessing import LabelEncoder

# for col in cols :
#   le = LabelEncoder()
#   train[col] = le.fit_transform(train[col])
#   test[col] = le.transform(test[col])

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)
# print(X_tr.shape, X_val.shape, y_tr.shape, y_val.shape)

# from sklearn.ensemble import RandomForestRegressor
# rf = RandomForestRegressor(random_state = 0)
# rf.fit(X_tr, y_tr)
# pred = rf.predict(X_val)
# print(pred)
# 937.577739684466

import lightgbm as lgb
model = lgb.LGBMRegressor(random_state = 0, verbose=-1)
model.fit(X_tr,y_tr)
pred = model.predict(X_val)
# 946.7024808036148

from sklearn.metrics import mean_absolute_error
print(mean_absolute_error(y_val, pred))

pred = model.predict(test)
submit = pd.DataFrame({'pred':pred})
submit.to_csv('result.csv',index = False)

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