mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-10-31 18:48:35 +00:00
69 lines
2.1 KiB
Markdown
69 lines
2.1 KiB
Markdown
|
---
|
||
|
|
title: ML Model Mixin
|
||
|
|
slug: /sdk/python/api-reference/mlmodel-mixin
|
||
|
|
---
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
[{% image align="right" style="float:right;" src="https://img.shields.io/badge/-source-cccccc?style=flat-square" /%}](https://github.com/open-metadata/OpenMetadata/tree/main/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py#L0")
|
||
|
|
|
||
|
|
# module `mlmodel_mixin`
|
||
|
|
Mixin class containing Lineage specific methods
|
||
|
|
|
||
|
|
To be used by OpenMetadata class
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
---
|
||
|
|
|
||
|
|
[{% image align="right" style="float:right;" src="https://img.shields.io/badge/-source-cccccc?style=flat-square" /%}](https://github.com/open-metadata/OpenMetadata/tree/main/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py#L49")
|
||
|
|
|
||
|
|
## class `OMetaMlModelMixin`
|
||
|
|
OpenMetadata API methods related to MlModel.
|
||
|
|
|
||
|
|
To be inherited by OpenMetadata
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
---
|
||
|
|
|
||
|
|
[{% image align="right" style="float:right;" src="https://img.shields.io/badge/-source-cccccc?style=flat-square" /%}](https://github.com/open-metadata/OpenMetadata/tree/main/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py#L58")
|
||
|
|
|
||
|
|
### method `add_mlmodel_lineage`
|
||
|
|
|
||
|
|
```python
|
||
|
|
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
|
||
|
|
|
||
|
|
---
|
||
|
|
|
||
|
|
[{% image align="right" style="float:right;" src="https://img.shields.io/badge/-source-cccccc?style=flat-square" /%}](https://github.com/open-metadata/OpenMetadata/tree/main/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py#L97")
|
||
|
|
|
||
|
|
### method `get_mlmodel_sklearn`
|
||
|
|
|
||
|
|
```python
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
---
|
||
|
|
|
||
|
|
|