Machine Learning cho Marketing với Python
Karolis Urbonas
Head of Analytics & Science, Amazon
Chỉ số chính:
R-squared - thước đo phần trăm phương sai được mô hình giải thích. Chỉ áp dụng cho hồi quy, không áp dụng cho phân loại. Càng cao càng tốt.
Giá trị p của hệ số - xác suất hệ số (hồi quy hoặc phân loại) xuất hiện do ngẫu nhiên. Càng thấp càng tốt. Ngưỡng thường dùng: 5% và 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())


Machine Learning cho Marketing với Python