lightgbm.rst 1.01 KB
Newer Older
1
2
3
4
lightgbm package
================


5
Data Structure API
6
7
------------------

8
.. autoclass:: lightgbm.Dataset
9
10
11
    :members:
    :show-inheritance:

12
.. autoclass:: lightgbm.Booster
13
14
15
    :members:
    :show-inheritance:

16
17
18
19
20
21
22
23
24
25

Training API
------------

.. autofunction:: lightgbm.train

.. autofunction:: lightgbm.cv


Scikit-learn API
26
27
----------------

28
.. autoclass:: lightgbm.LGBMModel
29
30
31
    :members:
    :show-inheritance:

32
.. autoclass:: lightgbm.LGBMClassifier
33
34
35
    :members:
    :show-inheritance:

36
.. autoclass:: lightgbm.LGBMRegressor
37
38
39
    :members:
    :show-inheritance:

40
.. autoclass:: lightgbm.LGBMRanker
41
42
43
44
    :members:
    :show-inheritance:


45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
Callbacks
---------

.. autofunction:: lightgbm.early_stopping

.. autofunction:: lightgbm.print_evaluation

.. autofunction:: lightgbm.record_evaluation

.. autofunction:: lightgbm.reset_parameter


Plotting
--------

.. autofunction:: lightgbm.plot_importance

.. autofunction:: lightgbm.plot_metric

.. autofunction:: lightgbm.plot_tree

.. autofunction:: lightgbm.create_tree_digraph
67