"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "a5f43c9d559848aed71a870321d2729fe64beab1"
__init__.py 919 Bytes
Newer Older
wxchan's avatar
wxchan committed
1
2
3
4
5
6
7
8
# coding: utf-8
"""LightGBM, Light Gradient Boosting Machine.

Contributors: https://github.com/Microsoft/LightGBM/graphs/contributors
"""

from __future__ import absolute_import

9
10
11
12
13
from .basic import Booster, Dataset
from .callback import (early_stopping, print_evaluation, record_evaluation,
                       reset_parameter)
from .engine import cv, train

wxchan's avatar
wxchan committed
14
15
16
17
try:
    from .sklearn import LGBMModel, LGBMRegressor, LGBMClassifier, LGBMRanker
except ImportError:
    pass
18
try:
19
    from .plotting import plot_importance, plot_metric, plot_tree, create_tree_digraph
20
21
except ImportError:
    pass
wxchan's avatar
wxchan committed
22
23


24
__version__ = 0.2
wxchan's avatar
wxchan committed
25
26
27

__all__ = ['Dataset', 'Booster',
           'train', 'cv',
28
           'LGBMModel', 'LGBMRegressor', 'LGBMClassifier', 'LGBMRanker',
29
           'print_evaluation', 'record_evaluation', 'reset_parameter', 'early_stopping',
30
           'plot_importance', 'plot_metric', 'plot_tree', 'create_tree_digraph']