我决定开始使用xgboost并使用单列车样本进行测试。
import xgboost as xgb
data=[[42]]
dtrain=xgb.DMatrix(data, label=[1])
xgb.train(dict(), dtrain).predict(dtrain)
#array([ 0.90156281], dtype=float32)
dtrain=xgb.DMatrix(data, label=[2])
xgb.train(dict(), dtrain).predict(strain)
#array([ 1.70468843], dtype=float32)
dtrain=xgb.DMatrix(data, label=[3])
xgb.train(dict(), dtrain).predict(strain)
#array([ 2.50781417], dtype=float32)
我应该改变什么来获取!单身!训练数据样本对相同测试数据的理想预测?
答案 0 :(得分:0)
好的,似乎xgboost使用的默认迭代次数太少了。只是作为参数传递num_round = 100修复它