PyCaret 回归

获取数据 Getting Data

%%time
from pycaret.datasets import get_data
boston = get_data('boston') 

环境配置 Setting up Environment

# 步骤1：导入模块 Step 1: Importing a Module 
from pycaret.regression import *

# 步骤2：初始化安装程序 Step 2: Initializing the setup
reg1 = setup(data = boston, target = 'medv')

模型比较 Compare Models

compare_models()

compare_models

默认值
fold K折交叉验证 10
round 四舍五入到的小数位数 4
sort 排序（高-低） R2

help(compare_models)

评估标准

Regression: MAE, MSE, RMSE, R2, RMSLE, MAPE

回归器

英	中
CatBoost Regressor	CatBoost回归器
Gradient Bosting Regressor	梯度增强回归器
Extra Trees Regressor	额外树回归器
Extreme Gradient Boosting	极梯度增强
Random Forest	随机森林
Light Gradient Boosting Machine	轻型梯度增压机
AdaBoost Regressor	AdaBoost回归器
Decision Tree	决策树
Linear Regression	线性回归
Ridge Regression	岭回归
Least Angle Regression	最小角度回归
Bayesian Ridge	贝叶斯岭
TheilSen Regressor	TheilSen回归器
Lasso Regression	Lasso回归
Elastic Net	弹性网
Random Sample Consensus	随机样本一致性
Orthogonal Matching Pursuit	正交匹配追踪
Huber Regressor	休伯回归器
K Neighbors Regressor	K近邻回归器
Support Vector Machine	支持向量机
Lasso Least Angle Regression	套索最小角度回归
Passive Aggressive Regressor	被动攻击回归器