wehub-resource-sync
32 lines
942 B
Python
32 lines
942 B
Python
from abc import abstractmethod
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING
|
|
|
|
from mlflow.data.evaluation_dataset import EvaluationDataset
|
|
|
|
if TYPE_CHECKING:
|
|
from mlflow.models.utils import PyFuncInput, PyFuncOutput
|
|
|
|
|
|
@dataclass
|
|
class PyFuncInputsOutputs:
|
|
inputs: list["PyFuncInput"]
|
|
outputs: list["PyFuncOutput"] | None = None
|
|
|
|
|
|
class PyFuncConvertibleDatasetMixin:
|
|
@abstractmethod
|
|
def to_pyfunc(self) -> PyFuncInputsOutputs:
|
|
"""
|
|
Converts the dataset to a collection of pyfunc inputs and outputs for model
|
|
evaluation. Required for use with mlflow.evaluate().
|
|
May not be implemented by all datasets.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def to_evaluation_dataset(self, path=None, feature_names=None) -> EvaluationDataset:
|
|
"""
|
|
Converts the dataset to an EvaluationDataset for model evaluation.
|
|
May not be implemented by all datasets.
|
|
"""
|