Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rewrite Dask interface. (#4819)

Browse Source
This commit is contained in:
Jiaming Yuan
2019-09-25 01:30:14 -04:00
committed by GitHub
parent 562bb0ae31
commit b8433c455a
17 changed files with 1002 additions and 361 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/__init__.py
View File

@@ -11,9 +11,9 @@ import os
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
from . import dask
try:
from .sklearn import XGBModel, XGBClassifier, XGBRegressor, XGBRanker
from .sklearn import XGBRFClassifier, XGBRFRegressor
@@ -30,4 +30,4 @@ __all__ = ['DMatrix', 'Booster',
'RabitTracker',
'XGBModel', 'XGBClassifier', 'XGBRegressor', 'XGBRanker',
'XGBRFClassifier', 'XGBRFRegressor',
'plot_importance', 'plot_tree', 'to_graphviz']
'plot_importance', 'plot_tree', 'to_graphviz', 'dask']