Expose Number of Trees in SW MOJO

Description

Allow users to get the number of trees (ntrees) in a Tree-Based model (i.e. GBM, DRF) when early stopping is enabled.

Note, if early stopping is enabled the algorithm can overwrite a user's ntree value. To access the model's actual ntree value, a user currently would have to use the mixed api (https://github.com/h2oai/h2o-droplets/blob/master/h2o-sw-mixed-api-droplet/src/main/scala/water/droplets/H2OSWMixedAPIDroplet.scala#L100).

Assignee

Marek Novotny

Reporter

Lauren DiPerna

Labels

None

CustomerVisible

No

testcase 1

None

testcase 2

None

testcase 3

None

h2ostream link

None

Affected Spark version

None

AffectedContact

None

AffectedCustomers

None

AffectedPilots

None

AffectedOpenSource

None

Support Assessment

None

Customer Request Type

None

Support ticket URL

None

End date

None

Baseline start date

None

Baseline end date

None

Task progress

None

Task mode

None

ReleaseNotesHidden

None

Fix versions

Priority

Major
Configure