cancel
Showing results for 
Search instead for 
Did you mean: 
Data Engineering
cancel
Showing results for 
Search instead for 
Did you mean: 

Set tags for an MLFlow Experiment using Python?

NAS
New Contributor III

There is this rest API: https://www.mlflow.org/docs/latest/rest-api.html#set-experiment-tag

Can I do the same from python's MLFlow API?

2 REPLIES 2

Kaniz
Community Manager
Community Manager

Hi @ NAS! My name is Kaniz, and I'm the technical moderator here. Great to meet you, and thanks for your question! Let's see if your peers in the community have an answer to your question first. Or else I will get back to you soon. Thanks.

NAS
New Contributor III

Someone answered first in StackOverflow. Here it is:

from mlflow.tracking import MlflowClient
 
# Create an experiment with a name that is unique and case sensitive.
client = MlflowClient()
experiment_id = client.create_experiment("Social NLP Experiments")
client.set_experiment_tag(experiment_id, "nlp.framework", "Spark NLP")

Welcome to Databricks Community: Lets learn, network and celebrate together

Join our fast-growing data practitioner and expert community of 80K+ members, ready to discover, help and collaborate together while making meaningful connections. 

Click here to register and join today! 

Engage in exciting technical discussions, join a group with your peers and meet our Featured Members.