2017-09-05 4 views
0

J'ai décidé de commencer à utiliser xgboost et l'ai testé avec un échantillon de train unique.xgboost mal à l'échantillon unique

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) 

Que dois-je changer pour obtenir! Unique! Prévision idéale de l'échantillon de données de train pour les mêmes données de test?

Répondre

0

Ok, il semble que le nombre d'itérations par défaut utilisé par xgboost est trop faible. Passer juste comme l'argument num_round = 100 le fixe