Skip to main content

​
module mlmodel_mixin

Mixin class containing Lineage specific methods To be used by OpenMetadata class

​
class OMetaMlModelMixin

OpenMetadata API methods related to MlModel. To be inherited by OpenMetadata

​
method add_mlmodel_lineage

add_mlmodel_lineage(
    model: MlModel,
    description: Optional[str] = None
) → Dict[str, Any]
Iterates over MlModel’s Feature Sources and add the lineage information. Args: model: MlModel containing EntityReferences description: Lineage description Returns: List of added lineage information

​
method get_mlmodel_sklearn

get_mlmodel_sklearn(
    name: str,
    model,
    description: Optional[str] = None,
    service_name: str = 'scikit-learn'
) → CreateMlModelRequest
Get an MlModel Entity instance from a scikit-learn model. Sklearn estimators all extend BaseEstimator. Args: name: MlModel name model: sklearn estimator description: MlModel description service_name: Service name to use when creating sklearn service Returns: OpenMetadata CreateMlModelRequest Entity