mirror of
https://github.com/Toumash/mlflow-docker
synced 2025-11-04 15:19:21 +01:00
23 lines
645 B
Python
23 lines
645 B
Python
import os
|
|
from random import random, randint
|
|
|
|
from mlflow import mlflow,log_metric, log_param, log_artifacts
|
|
|
|
if __name__ == "__main__":
|
|
with mlflow.start_run() as run:
|
|
mlflow.set_tracking_uri('https://mlflow.localhost')
|
|
print("Running mlflow_tracking.py")
|
|
|
|
log_param("param1", randint(0, 100))
|
|
|
|
log_metric("foo", random())
|
|
log_metric("foo", random() + 1)
|
|
log_metric("foo", random() + 2)
|
|
|
|
if not os.path.exists("outputs"):
|
|
os.makedirs("outputs")
|
|
with open("outputs/test.txt", "w") as f:
|
|
f.write("hello world!")
|
|
|
|
log_artifacts("outputs")
|