[python package] include dmlc-tracker into xgb python pkg (#4731)

This commit is contained in:
Xu Xiao
2019-08-06 00:21:07 +08:00
committed by Jiaming Yuan
parent b68de018b8
commit 97eece6ea0
6 changed files with 361 additions and 17 deletions

View File

@@ -12,6 +12,8 @@ from .core import DMatrix, Booster
from .training import train, cv
from . import rabit # noqa
from . import dask # noqa
from . import tracker # noqa
from .tracker import RabitTracker # noqa
try:
from .sklearn import XGBModel, XGBClassifier, XGBRegressor, XGBRanker
from .sklearn import XGBRFClassifier, XGBRFRegressor
@@ -25,6 +27,7 @@ with open(VERSION_FILE) as f:
__all__ = ['DMatrix', 'Booster',
'train', 'cv',
'RabitTracker',
'XGBModel', 'XGBClassifier', 'XGBRegressor', 'XGBRanker',
'XGBRFClassifier', 'XGBRFRegressor',
'plot_importance', 'plot_tree', 'to_graphviz']