2017-09-05 45 views
0

我決定開始使用xgboost並使用單個樣本進行測試。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

好吧,似乎像xgboost使用的默認迭代次數太少。只是作爲參數傳遞num_round = 100修復它