Python | RabbidsX

假设有这样一个模型：

# coding: utf-8

import numpy as np
import pandas as pd
import xgboost as xgb

# 构造模型
model = xgb.XGBRegressor(
    learning_rate=0.2,
    n_estimators=2000,
    max_depth=10,
    min_child_weight=1,
    gamma=8,
    subsample=0.8,
    colsample_bytree=0.8,
    objective="reg:linear",
    booster="gbtree",
    n_jobs=4,
    scale_pos_weight=1,
    random_state=27
)

# 训练
model.fit(X_train, Y_train)

# 预测
Y_pred = model.predict(X_test)

# coding: utf-8

import numpy as np

import pandas as pd

import xgboost as xgb

# 构造模型

model = xgb.XGBRegressor(

learning_rate=0.2,

n_estimators=2000,

max_depth=10,

min_child_weight=1,

gamma=8,

subsample=0.8,

colsample_bytree=0.8,

objective="reg:linear",

booster="gbtree",

n_jobs=4,

scale_pos_weight=1,

random_state=27

)

# 训练

model.fit(X_train, Y_train)

# 预测

Y_pred = model.predict(X_test)

阅读全文

分类 Python 下的文章

XGBoost的保存模型、加载模型、继续训练

Pandas获取网页上的表格内容

最近

分类