python实现xgboost回归,请给出实例
时间: 2023-08-02 18:08:10 浏览: 92
xgboost算法的python实现
好的,以下是一个简单的 XGBoost 回归模型的 Python 实现例子:
首先,我们需要安装 XGBoost 库。可以通过 pip 命令安装:
```
pip install xgboost
```
接下来,我们准备数据并将其分成训练集和测试集:
```python
import pandas as pd
from sklearn.model_selection import train_test_split
# 读取数据
data = pd.read_csv('data.csv')
# 分离目标变量和特征变量
X = data.drop('target', axis=1)
y = data['target']
# 分割数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
```
然后,我们可以创建 XGBoost 模型并对其进行训练:
```python
import xgboost as xgb
# 定义模型参数
params = {
'objective': 'reg:squarederror',
'max_depth': 3,
'learning_rate': 0.1,
'n_estimators': 100
}
# 创建 XGBoost 模型
model = xgb.XGBRegressor(**params)
# 训练模型
model.fit(X_train, y_train)
```
最后,我们可以用测试集对模型进行评估:
```python
from sklearn.metrics import mean_squared_error
# 使用模型进行预测
y_pred = model.predict(X_test)
# 计算均方根误差
rmse = mean_squared_error(y_test, y_pred, squared=False)
print('RMSE:', rmse)
```
这就是一个简单的 XGBoost 回归模型的 Python 实现例子。
阅读全文