Define export symbols in callback module. (#7665)

This commit is contained in:
Jiaming Yuan 2022-02-19 18:52:41 +08:00 committed by GitHub
parent 7366d3b20c
commit b76c5d54bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,6 +19,14 @@ from .core import Booster, DMatrix, XGBoostError, _get_booster_layer_trees
from .compat import STRING_TYPES
__all__ = [
"TrainingCallback",
"LearningRateScheduler",
"EarlyStopping",
"EvaluationMonitor",
"TrainingCheckPoint",
]
_Score = Union[float, Tuple[float, float]]
_ScoreList = Union[List[float], List[Tuple[float, float]]]