mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-10-30 18:17:53 +00:00
2.1 KiB
2.1 KiB
| title | slug |
|---|---|
| ML Model Mixin | /sdk/python/api-reference/mlmodel-mixin |
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. :param model: MlModel containing EntityReferences :param description: Lineage description :return: 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. :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