dmlc / dmlc/xgboost

[jvm-packages] return the best model after early stop

Open
#10,816 3 comments 0 reactions 0 assignees View on GitHub
feature-request
Dominant language
C++
Stars
28.8k
Forks
8.9k
Avg merge
1d 12h
Merged PRs (30d)
54

Description

Does xgboost4j-spark support to get the best model after early stop?

It seems it will get the model at that iteration which is best iteration + num_early_stopping_rounds, am i wrong?
how could i get the best model?
image

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by examining the xgboost4j-spark early-stopping behavior and the model-retrieval API mentioned in the issue. Reproduce whether the returned model is from best iteration or a later iteration, then establish and document the expected way to obtain the best model.

Written by the indexing model from the issue text.

Assessment

Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.