https://kedro.org/ logo
#questions
Title
# questions
j

Júlio Resende

03/11/2024, 8:12 PM
Hey guys! I'm new to Databricks and trying to figure out the best way to register my model in Databricks using Kedro. I don't want to serve my models as an API. I would just like to register them on the page below. Should I use MLflowModelTrackingDataset?
d

Deepyaman Datta

03/11/2024, 8:58 PM
I don't really use this in Databricks, but seems right. What does the Databricks documentation says? If it says to use
log_model
, then that's right, since that's essentially what the tracking dataset does.
👍 2
y

Yolan Honoré-Rougé

03/12/2024, 6:49 AM
Yes this dataset is made for this. You can pass
Copy code
save_args:
     registered_model_name: "my-model
To automatically register your dataset each time you save it.
👍 1
j

Júlio Resende

03/12/2024, 1:23 PM
After looking deeper, I think that the
MlflowArtifactDataset
+
PickleDataset
will be more appropriate for me, because my models are actually of a very unusual type. I'll carry out some tests and then leave some feedback in this thread. Thanks @Deepyaman Datta and @Yolan Honoré-Rougé
👍 1
2 Views