compat.py 4.26 KB
Newer Older
wxchan's avatar
wxchan committed
1
# coding: utf-8
2
"""Compatibility library."""
3

wxchan's avatar
wxchan committed
4
5
"""pandas"""
try:
6
    from pandas import DataFrame as pd_DataFrame
7
8
    from pandas import Series as pd_Series
    from pandas import concat
9
    from pandas.api.types import is_sparse as is_dtype_sparse
10
    PANDAS_INSTALLED = True
wxchan's avatar
wxchan committed
11
except ImportError:
12
13
    PANDAS_INSTALLED = False

14
    class pd_Series:  # type: ignore
15
16
        """Dummy class for pandas.Series."""

wxchan's avatar
wxchan committed
17
18
        pass

19
    class pd_DataFrame:  # type: ignore
20
21
        """Dummy class for pandas.DataFrame."""

wxchan's avatar
wxchan committed
22
23
        pass

24
    concat = None
25
26
    is_dtype_sparse = None

27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""matplotlib"""
try:
    import matplotlib
    MATPLOTLIB_INSTALLED = True
except ImportError:
    MATPLOTLIB_INSTALLED = False

"""graphviz"""
try:
    import graphviz
    GRAPHVIZ_INSTALLED = True
except ImportError:
    GRAPHVIZ_INSTALLED = False

41
42
"""datatable"""
try:
43
44
    import datatable
    if hasattr(datatable, "Frame"):
45
        dt_DataTable = datatable.Frame
46
    else:
47
        dt_DataTable = datatable.DataTable
48
49
50
51
    DATATABLE_INSTALLED = True
except ImportError:
    DATATABLE_INSTALLED = False

52
    class dt_DataTable:  # type: ignore
53
        """Dummy class for datatable.DataTable."""
54
55
56
57

        pass


wxchan's avatar
wxchan committed
58
59
"""sklearn"""
try:
60
    from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin
wxchan's avatar
wxchan committed
61
    from sklearn.preprocessing import LabelEncoder
62
    from sklearn.utils.class_weight import compute_sample_weight
63
    from sklearn.utils.multiclass import check_classification_targets
64
    from sklearn.utils.validation import assert_all_finite, check_array, check_X_y
wxchan's avatar
wxchan committed
65
    try:
66
        from sklearn.exceptions import NotFittedError
67
        from sklearn.model_selection import GroupKFold, StratifiedKFold
wxchan's avatar
wxchan committed
68
    except ImportError:
69
        from sklearn.cross_validation import GroupKFold, StratifiedKFold
70
        from sklearn.utils.validation import NotFittedError
71
72
73
74
75
76
77
78
79
80
    try:
        from sklearn.utils.validation import _check_sample_weight
    except ImportError:
        from sklearn.utils.validation import check_consistent_length

        # dummy function to support older version of scikit-learn
        def _check_sample_weight(sample_weight, X, dtype=None):
            check_consistent_length(sample_weight, X)
            return sample_weight

wxchan's avatar
wxchan committed
81
    SKLEARN_INSTALLED = True
82
83
84
85
86
87
88
89
90
    _LGBMModelBase = BaseEstimator
    _LGBMRegressorBase = RegressorMixin
    _LGBMClassifierBase = ClassifierMixin
    _LGBMLabelEncoder = LabelEncoder
    LGBMNotFittedError = NotFittedError
    _LGBMStratifiedKFold = StratifiedKFold
    _LGBMGroupKFold = GroupKFold
    _LGBMCheckXY = check_X_y
    _LGBMCheckArray = check_array
91
    _LGBMCheckSampleWeight = _check_sample_weight
92
    _LGBMAssertAllFinite = assert_all_finite
93
    _LGBMCheckClassificationTargets = check_classification_targets
94
    _LGBMComputeSampleWeight = compute_sample_weight
wxchan's avatar
wxchan committed
95
96
except ImportError:
    SKLEARN_INSTALLED = False
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

    class _LGBMModelBase:  # type: ignore
        """Dummy class for sklearn.base.BaseEstimator."""

        pass

    class _LGBMClassifierBase:  # type: ignore
        """Dummy class for sklearn.base.ClassifierMixin."""

        pass

    class _LGBMRegressorBase:  # type: ignore
        """Dummy class for sklearn.base.RegressorMixin."""

        pass

113
114
115
116
117
118
    _LGBMLabelEncoder = None
    LGBMNotFittedError = ValueError
    _LGBMStratifiedKFold = None
    _LGBMGroupKFold = None
    _LGBMCheckXY = None
    _LGBMCheckArray = None
119
    _LGBMCheckSampleWeight = None
120
    _LGBMAssertAllFinite = None
121
    _LGBMCheckClassificationTargets = None
122
    _LGBMComputeSampleWeight = None
123
124
125

"""dask"""
try:
126
127
    from dask import delayed
    from dask.array import Array as dask_Array
128
129
    from dask.dataframe import DataFrame as dask_DataFrame
    from dask.dataframe import Series as dask_Series
130
    from dask.distributed import Client, default_client, wait
131
132
133
    DASK_INSTALLED = True
except ImportError:
    DASK_INSTALLED = False
134

135
136
137
138
    delayed = None
    default_client = None
    wait = None

139
140
141
142
143
    class Client:  # type: ignore
        """Dummy class for dask.distributed.Client."""

        pass

144
    class dask_Array:  # type: ignore
145
146
147
148
        """Dummy class for dask.array.Array."""

        pass

149
    class dask_DataFrame:  # type: ignore
150
151
152
153
        """Dummy class for dask.dataframe.DataFrame."""

        pass

154
    class dask_Series:  # type: ignore
155
        """Dummy class for dask.dataframe.Series."""
156
157

        pass