File size: 395 Bytes
473c3a0 |
1 2 3 4 5 6 7 8 9 10 11 |
from distiller.model2vec.utils import get_package_extras, importable
_REQUIRED_EXTRA = "inference"
for extra_dependency in get_package_extras("model2vec", _REQUIRED_EXTRA):
importable(extra_dependency, _REQUIRED_EXTRA)
from distiller.model2vec.inference.model import StaticModelPipeline, evaluate_single_or_multi_label
__all__ = ["StaticModelPipeline", "evaluate_single_or_multi_label"]
|