Python으로 배우는 마케팅용 Machine Learning
Karolis Urbonas
Head of Analytics & Science, Amazon
핵심 지표:
R-squared: 모형이 설명하는 분산 비율(%)을 나타내는 통계량. 회귀에만 적용, 분류에는 적용되지 않음. 높을수록 좋음.
계수 p-값: 회귀(또는 분류) 계수가 우연에 의해 관측될 확률. 낮을수록 좋음. 일반 임계값은 5%, 10%.
# Import the linear regression module from sklearn.linear_model import LinearRegression# Initialize the regression instance linreg = LinearRegression()# Fit model on the training data linreg.fit(train_X, train_Y)# Predict values on both training and testing data train_pred_Y = linreg.predict(train_X) test_pred_Y = linreg.predict(test_X)
# Import performance measurement functions from sklearn.metrics import mean_absolute_error from sklearn.metrics import mean_squared_error# Calculate metrics for training data rmse_train = np.sqrt(mean_squared_error(train_Y, train_pred_Y)) mae_train = mean_absolute_error(train_Y, train_pred_Y)# Calculate metrics for testing data rmse_test = np.sqrt(mean_squared_error(test_Y, test_pred_Y)) mae_test = mean_absolute_error(test_Y, test_pred_Y)# Print performance metrics print('RMSE train: {:.3f}; RMSE test: {:.3f}\nMAE train: {:.3f}, MAE test: {:.3f}'.format( rmse_train, rmse_test, mae_train, mae_test))
RMSE train: 0.717; RMSE test: 1.216
MAE train: 0.514, MAE test: 0.555
statsmodels 라이브러리 소개# Import the library import statsmodels.api as sm# Convert target variable to `numpy` array train_Y = np.array(train_Y)# Initialize and fit the model olsreg = sm.OLS(train_Y, train_X) olsreg = olsreg.fit()# Print model summary print(olsreg.summary())


Python으로 배우는 마케팅용 Machine Learning