Commit a0d7313b authored by Nikita Titov's avatar Nikita Titov Committed by Guolin Ke
Browse files

fixed docstrings (#2451)

parent 7b2963d9
......@@ -357,8 +357,8 @@ class _InnerPredictor(object):
Not exposed to user.
Used only for prediction, usually used for continued training.
Note
----
.. note::
Can be converted from Booster, but cannot be converted to Booster.
"""
......@@ -1939,8 +1939,8 @@ class Booster(object):
def __boost(self, grad, hess):
"""Boost Booster for one iteration with customized gradient statistics.
Note
----
.. note::
For multi-class task, the score is group by class_id first, then group by row_id.
If you want to get i-th row score in j-th class, the access way is score[j * num_data + i]
and you should group grad and hess in this way as well.
......@@ -2340,8 +2340,8 @@ class Booster(object):
pred_contrib : bool, optional (default=False)
Whether to predict feature contributions.
Note
----
.. note::
If you want to get more explanations for your model's predictions using SHAP values,
like SHAP interaction values,
you can install the shap package (https://github.com/slundberg/shap).
......@@ -2526,8 +2526,8 @@ class Booster(object):
If int, interpreted as index.
If string, interpreted as name.
Note
----
.. warning::
Categorical features are not supported.
bins : int, string or None, optional (default=None)
......
......@@ -109,8 +109,8 @@ def record_evaluation(eval_result):
def reset_parameter(**kwargs):
"""Create a callback that resets the parameter after the first iteration.
Note
----
.. note::
The initial parameter will still take in-effect on first iteration.
Parameters
......@@ -154,8 +154,6 @@ def reset_parameter(**kwargs):
def early_stopping(stopping_rounds, first_metric_only=False, verbose=True):
"""Create a callback that activates early stopping.
Note
----
Activates early stopping.
The model will train until the validation score stops improving.
Validation score needs to improve at least every ``early_stopping_rounds`` round(s)
......
......@@ -101,8 +101,8 @@ def train(params, train_set, num_boost_round=100,
evals_result: dict or None, optional (default=None)
This dictionary used to store all evaluation results of all the items in ``valid_sets``.
Example
-------
.. rubric:: Example
With a ``valid_sets`` = [valid_set, train_set],
``valid_names`` = ['eval', 'train']
and a ``params`` = {'metric': 'logloss'}
......@@ -115,8 +115,8 @@ def train(params, train_set, num_boost_round=100,
If int, the eval metric on the valid set is printed at every ``verbose_eval`` boosting stage.
The last boosting stage or the boosting stage found by using ``early_stopping_rounds`` is also printed.
Example
-------
.. rubric:: Example
With ``verbose_eval`` = 4 and at least one item in ``valid_sets``,
an evaluation metric is printed every 4 (instead of 1) boosting stages.
......
......@@ -469,8 +469,8 @@ def create_tree_digraph(booster, tree_index=0, show_info=None, precision=3,
old_node_attr=None, old_edge_attr=None, old_body=None, old_strict=False, **kwargs):
"""Create a digraph representation of specified tree.
Note
----
.. note::
For more information please visit
https://graphviz.readthedocs.io/en/stable/api.html#digraph.
......@@ -545,8 +545,8 @@ def plot_tree(booster, ax=None, tree_index=0, figsize=None,
show_info=None, precision=3, **kwargs):
"""Plot specified tree.
Note
----
.. note::
It is preferable to use ``create_tree_digraph()`` because of its lossless quality
and returned objects can be also rendered and displayed directly inside a Jupyter notebook.
......
......@@ -40,8 +40,8 @@ class _ObjectiveFunctionWrapper(object):
hess : array-like of shape = [n_samples] or shape = [n_samples * n_classes] (for multi-class task)
The value of the second order derivative (Hessian) for each sample point.
Note
----
.. note::
For multi-class task, the y_pred is group by class_id first, then group by row_id.
If you want to get i-th row y_pred in j-th class, the access way is y_pred[j * num_data + i]
and you should group grad and hess in this way as well.
......@@ -127,8 +127,8 @@ class _EvalFunctionWrapper(object):
is_higher_better : bool
Is eval result higher better, e.g. AUC is ``is_higher_better``.
Note
----
.. note::
For multi-class task, the y_pred is group by class_id first, then group by row_id.
If you want to get i-th row y_pred in j-th class, the access way is y_pred[j * num_data + i].
"""
......@@ -244,8 +244,8 @@ class LGBMModel(_LGBMModelBase):
Other parameters for the model.
Check http://lightgbm.readthedocs.io/en/latest/Parameters.html for more parameters.
Note
----
.. warning::
\*\*kwargs is not supported in sklearn, it may cause unexpected issues.
Attributes
......@@ -421,8 +421,8 @@ class LGBMModel(_LGBMModelBase):
If int, the eval metric on the eval set is printed at every ``verbose`` boosting stage.
The last boosting stage or the boosting stage found by using ``early_stopping_rounds`` is also printed.
Example
-------
.. rubric:: Example
With ``verbose`` = 4 and at least one item in ``eval_set``,
an evaluation metric is printed every 4 (instead of 1) boosting stages.
......@@ -626,8 +626,8 @@ class LGBMModel(_LGBMModelBase):
pred_contrib : bool, optional (default=False)
Whether to predict feature contributions.
Note
----
.. note::
If you want to get more explanations for your model's predictions using SHAP values,
like SHAP interaction values,
you can install the shap package (https://github.com/slundberg/shap).
......@@ -705,8 +705,8 @@ class LGBMModel(_LGBMModelBase):
def feature_importances_(self):
"""Get feature importances.
Note
----
.. note::
Feature importance in sklearn interface used to normalize to 1,
it's deprecated after 2.0.4 and is the same as Booster.feature_importance() now.
``importance_type`` attribute is passed to the function
......@@ -834,8 +834,8 @@ class LGBMClassifier(LGBMModel, _LGBMClassifierBase):
pred_contrib : bool, optional (default=False)
Whether to predict feature contributions.
Note
----
.. note::
If you want to get more explanations for your model's predictions using SHAP values,
like SHAP interaction values,
you can install the shap package (https://github.com/slundberg/shap).
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment