__init__.py 531 Bytes
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
6
7
8
9
10
11
12
13
# coding: utf-8
"""LightGBM, Light Gradient Boosting Machine.

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

from __future__ import absolute_import

import os

from .basic import Predictor, Dataset, Booster
from .engine import train, cv
try:
14
    from .sklearn import LGBMModel, LGBMRegressor, LGBMClassifier, LGBMRanker
Guolin Ke's avatar
Guolin Ke committed
15
16
17
except ImportError:
    pass

Guolin Ke's avatar
Guolin Ke committed
18
19

__version__ = 0.1
Guolin Ke's avatar
Guolin Ke committed
20
21
22

__all__ = ['Dataset', 'Booster',
           'train', 'cv',
Guolin Ke's avatar
Guolin Ke committed
23
           'LGBMModel','LGBMRegressor', 'LGBMClassifier', 'LGBMRanker']