668b8a0ea4
* Switch from rabit to the collective communicator * fix size_t specialization * really fix size_t * try again * add include * more include * fix lint errors * remove rabit includes * fix pylint error * return dict from communicator context * fix communicator shutdown * fix dask test * reset communicator mocklist * fix distributed tests * do not save device communicator * fix jvm gpu tests * add python test for federated communicator * Update gputreeshap submodule Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
68 lines
1.2 KiB
Python
68 lines
1.2 KiB
Python
"""XGBoost: eXtreme Gradient Boosting library.
|
|
|
|
Contributors: https://github.com/dmlc/xgboost/blob/master/CONTRIBUTORS.md
|
|
"""
|
|
|
|
from . import tracker # noqa
|
|
from . import collective, dask
|
|
from .core import (
|
|
Booster,
|
|
DataIter,
|
|
DeviceQuantileDMatrix,
|
|
DMatrix,
|
|
QuantileDMatrix,
|
|
_py_version,
|
|
build_info,
|
|
)
|
|
from .tracker import RabitTracker # noqa
|
|
from .training import cv, train
|
|
|
|
try:
|
|
from .config import config_context, get_config, set_config
|
|
from .plotting import plot_importance, plot_tree, to_graphviz
|
|
from .sklearn import (
|
|
XGBClassifier,
|
|
XGBModel,
|
|
XGBRanker,
|
|
XGBRegressor,
|
|
XGBRFClassifier,
|
|
XGBRFRegressor,
|
|
)
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
__version__ = _py_version()
|
|
|
|
|
|
__all__ = [
|
|
# core
|
|
"DMatrix",
|
|
"DeviceQuantileDMatrix",
|
|
"QuantileDMatrix",
|
|
"Booster",
|
|
"DataIter",
|
|
"train",
|
|
"cv",
|
|
# utilities
|
|
"RabitTracker",
|
|
"build_info",
|
|
"plot_importance",
|
|
"plot_tree",
|
|
"to_graphviz",
|
|
"set_config",
|
|
"get_config",
|
|
"config_context",
|
|
# sklearn
|
|
"XGBModel",
|
|
"XGBClassifier",
|
|
"XGBRegressor",
|
|
"XGBRanker",
|
|
"XGBRFClassifier",
|
|
"XGBRFRegressor",
|
|
# dask
|
|
"dask",
|
|
# collective
|
|
"collective",
|
|
]
|