You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
If early stopping is enabled, GBM can reset the ntree value. Let's add a new method to get the actual ntree value (whether CV is enabled or not), rather than the original ntree value a user set before building their model.
Currently if you want the actual ntree value you need to do:
Sebastien Poirier commented: actually, there are currently simpler access to those properties than what is mentioned in the description.
at least for Python:
{{model.summary()['number_of_trees']}}
If early stopping is enabled, GBM can reset the ntree value. Let's add a new method to get the actual ntree value (whether CV is enabled or not), rather than the original ntree value a user set before building their model.
Currently if you want the actual ntree value you need to do:
Python:
model._model_json['output']['model_summary']['number_of_trees']
R:
my_model@model$model_summary['number_of_trees']
we could add a method like:
model.ntree_actual
The text was updated successfully, but these errors were encountered: