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

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 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