mlflow_tracking.py
1 import os 2 from random import randint, random 3 4 from mlflow import log_artifacts, log_metric, log_param 5 6 if __name__ == "__main__": 7 print("Running mlflow_tracking.py") 8 9 log_param("param1", randint(0, 100)) 10 11 log_metric("foo", random()) 12 log_metric("foo", random() + 1) 13 log_metric("foo", random() + 2) 14 15 if not os.path.exists("outputs"): 16 os.makedirs("outputs") 17 with open("outputs/test.txt", "w") as f: 18 f.write("hello world!") 19 20 log_artifacts("outputs")