qcode
  • 创作工场专业版,一问多答!更智能、更完整、更原创
  • 这是一个超强内容创作平台,它会写文章、做设计、绘图、敲代码,有问必答!
🔥点击使用-高级专业版🔥

写一个XGboost模型

时间2023-04-10 16:35:54 提问来自 内蒙古自治区赤峰市 的网友 分类创作 浏览216
```python import xgboost as xgb from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split import numpy as np # 加载数据集,切分训练集和测试集,占比7:3 boston = load_boston() # 获取波士顿房价数据集 X, y = boston.data, boston.target # 分割特征和目标变量 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3) # 切分训练集和测试集,占比7:3 # XGBoost训练过程 xgbregressor = xgb.XGBRegressor(maxdepth=5, learningrate=0.1) # 构造XGboost回归模型对象 xgbregressor.fit(X=X_train,y=y_train) # 训练模型 preds = xgbregressor.predict(X=X) # 预测 print("RMSE on test set : {}".format(np.sqrt(mean((preds - y)**2)))) # 打印RMSE评估结果

点击这里,进行内容创作

创作工场 免责声明:
以上内容除特别注明外均来源于网友提问,创作工场回答,未经许可,严谨转载。
点击这里>>使用🔥专业版,更聪明、更完整、更原创!

上一篇 mapCoordinateList: [ 下一篇 能否再详细一点,包括历史方面,谢谢!