SIGN IN SIGN UP
dmlc / xgboost UNCLAIMED

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

2015-08-01 11:27:13 -07:00
Python API Reference
====================
This page gives the Python API reference of xgboost, please also refer to Python Package Introduction for more information about the Python package.
2015-08-01 13:47:41 -07:00
.. contents::
:backlinks: none
:local:
Global Configuration
--------------------
.. autofunction:: xgboost.config_context
.. autofunction:: xgboost.set_config
.. autofunction:: xgboost.get_config
.. autofunction:: xgboost.build_info
2015-08-01 11:27:13 -07:00
Core Data Structure
-------------------
.. automodule:: xgboost.core
.. autoclass:: xgboost.DMatrix
:members:
:show-inheritance:
.. autoclass:: xgboost.QuantileDMatrix
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.ExtMemQuantileDMatrix
:members:
:inherited-members:
:show-inheritance:
2015-08-01 11:27:13 -07:00
.. autoclass:: xgboost.Booster
:members:
:show-inheritance:
:special-members: __getitem__
2015-08-01 11:27:13 -07:00
.. autoclass:: xgboost.DataIter
:members:
:show-inheritance:
2015-08-01 11:27:13 -07:00
.. autoclass:: xgboost.core.Categories
2015-08-01 11:27:13 -07:00
Learning API
------------
.. automodule:: xgboost.training
.. autofunction:: xgboost.train
.. autofunction:: xgboost.cv
Scikit-Learn API
----------------
2015-08-01 11:27:13 -07:00
.. automodule:: xgboost.sklearn
.. autoclass:: xgboost.XGBRegressor
:members:
:inherited-members:
2015-08-01 11:27:13 -07:00
:show-inheritance:
.. autoclass:: xgboost.XGBClassifier
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.XGBRanker
:members:
:inherited-members:
2015-08-01 11:27:13 -07:00
:show-inheritance:
2019-05-25 18:30:24 +12:00
.. autoclass:: xgboost.XGBRFRegressor
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.XGBRFClassifier
:members:
:inherited-members:
:show-inheritance:
Plotting API
------------
.. automodule:: xgboost.plotting
.. autofunction:: xgboost.plot_importance
.. autofunction:: xgboost.plot_tree
.. autofunction:: xgboost.to_graphviz
.. _callback_api:
Callback API
------------
.. automodule:: xgboost.callback
.. autoclass:: xgboost.callback.TrainingCallback
:members:
.. autoclass:: xgboost.callback.EvaluationMonitor
:members:
:show-inheritance:
.. autoclass:: xgboost.callback.EarlyStopping
:members:
:show-inheritance:
.. autoclass:: xgboost.callback.LearningRateScheduler
:members:
:show-inheritance:
2020-11-12 17:51:32 +08:00
.. autoclass:: xgboost.callback.TrainingCheckPoint
:members:
:show-inheritance:
.. _dask_api:
Dask API
--------
.. automodule:: xgboost.dask
.. autoclass:: xgboost.dask.DaskDMatrix
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.dask.DaskQuantileDMatrix
:members:
:inherited-members:
:show-inheritance:
.. autofunction:: xgboost.dask.train
2019-09-25 01:30:14 -04:00
.. autofunction:: xgboost.dask.predict
2020-11-12 17:51:32 +08:00
.. autofunction:: xgboost.dask.inplace_predict
.. autoclass:: xgboost.dask.DaskXGBClassifier
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.dask.DaskXGBRegressor
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.dask.DaskXGBRanker
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.dask.DaskXGBRFRegressor
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.dask.DaskXGBRFClassifier
:members:
:inherited-members:
:show-inheritance:
PySpark API
-----------
.. automodule:: xgboost.spark
.. autoclass:: xgboost.spark.SparkXGBClassifier
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.spark.SparkXGBClassifierModel
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.spark.SparkXGBRegressor
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.spark.SparkXGBRegressorModel
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.spark.SparkXGBRanker
:members:
:inherited-members:
:show-inheritance:
.. autoclass:: xgboost.spark.SparkXGBRankerModel
:members:
:inherited-members:
:show-inheritance:
2024-11-15 18:23:32 +08:00
Collective
----------
.. automodule:: xgboost.collective
.. autoclass:: xgboost.collective.Config
.. autofunction:: xgboost.collective.init
.. autofunction:: xgboost.collective.finalize
.. autofunction:: xgboost.collective.get_rank
.. autofunction:: xgboost.collective.get_world_size
.. autoclass:: xgboost.collective.CommunicatorContext
2024-11-15 18:23:32 +08:00
.. automodule:: xgboost.tracker
.. autoclass:: xgboost.tracker.RabitTracker