2017-11-07 15 views
0

xgb.train을 사용하는 경우 내 모델의 트리 수를 어떻게 지정할 수 있습니까?xgb.train을 사용하여 내 xgboost 모델의 트리 수를 어떻게 지정할 수 있습니까?

xgboost.XGBRegressor()에서 'n_estimators'매개 변수를 사용할 수 있지만 xgb.train()에서 무엇을해야합니까?

전 Google을 검색했지만 어떤 답변도 찾지 못했습니다. 미리 감사드립니다.

+0

안녕, 당신이 해결 않았다 당신의 문제? –

+0

@VivekKumar 네, 고마워요! – Statham

답변

1

num_boost_round. api documentation에 명시된 바와 같이 :

num_boost_round (int) - 부스팅 반복 수입니다.

정확히 당신이 여기 내 다른 대답을 볼 수 xgb.train 및 scikit 래퍼 (XGBRegressor 또는 XGBClassifier)의 출력을 일치 시키려면 :