mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-04 04:39:10 +00:00
20 lines
399 B
Python
20 lines
399 B
Python
from datahub.metadata.urns import MlModelGroupUrn
|
|
from datahub.sdk import DataHubClient
|
|
from datahub.sdk.mlmodel import MLModel
|
|
|
|
client = DataHubClient.from_env()
|
|
|
|
model = MLModel(
|
|
id="my-recommendations-model",
|
|
platform="mlflow",
|
|
)
|
|
|
|
model.set_model_group(
|
|
MlModelGroupUrn(
|
|
platform="mlflow",
|
|
name="my-recommendations-model-group",
|
|
)
|
|
)
|
|
|
|
client.entities.upsert(model)
|