2017-09-05 4 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

Ok、xgboostで使用されるデフォルトの反復回数が少なすぎるようです。引数num_round = 100を渡すだけで修正されます

関連する問題