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. :param model: MlModel containing EntityReferences :param description: Lineage description :return: List of added lineage information

method get_mlmodel_sklearn

Get an MlModel Entity instance from a scikit-learn model.

Sklearn estimators all extend BaseEstimator. :param name: MlModel name :param model: sklearn estimator :param description: MlModel description :param service_name: Service name to use when creating sklearn service :return: OpenMetadata CreateMlModelRequest Entity