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