forked from mlflow/mlflow
-
Notifications
You must be signed in to change notification settings - Fork 1
/
__init__.py
50 lines (42 loc) · 1.31 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
"""
The ``mlflow.models`` module provides an API for saving machine learning models in
"flavors" that can be understood by different downstream tools.
The built-in flavors are:
- :py:mod:`mlflow.pyfunc`
- :py:mod:`mlflow.h2o`
- :py:mod:`mlflow.keras`
- :py:mod:`mlflow.lightgbm`
- :py:mod:`mlflow.pytorch`
- :py:mod:`mlflow.sklearn`
- :py:mod:`mlflow.spark`
- :py:mod:`mlflow.statsmodels`
- :py:mod:`mlflow.tensorflow`
- :py:mod:`mlflow.xgboost`
- :py:mod:`mlflow.spacy`
- :py:mod:`mlflow.fastai`
- :py:mod:`mlflow.paddle`
For details, see `MLflow Models <../models.html>`_.
"""
from .model import Model
from .flavor_backend import FlavorBackend
from ..utils.environment import infer_pip_requirements
from .evaluation import evaluate, EvaluationDataset
__all__ = [
"Model",
"FlavorBackend",
"infer_pip_requirements",
"evaluate",
"EvaluationDataset",
]
# Under skinny-mlflow requirements, the following packages cannot be imported
# because of lack of numpy/pandas library, so wrap them with try...except block
try:
from .signature import ModelSignature, infer_signature # pylint: disable=unused-import
from .utils import ModelInputExample # pylint: disable=unused-import
__all__ += [
"ModelSignature",
"ModelInputExample",
"infer_signature",
]
except ImportError:
pass