mirror of
https://github.com/Toumash/mlflow-docker
synced 2025-11-04 23:29:19 +01:00
23 lines
625 B
Python
23 lines
625 B
Python
|
import os
|
||
|
|
from random import random, randint
|
||
|
|
|
||
|
|
import mlflow
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with mlflow.start_run() as run:
|
||
|
|
mlflow.set_tracking_uri('http://mlflow:5000')
|
||
|
|
print("Running mlflow_tracking.py")
|
||
|
|
|
||
|
|
mlflow.log_param("param1", randint(0, 100))
|
||
|
|
|
||
|
|
mlflow.log_metric("foo", random())
|
||
|
|
mlflow.log_metric("foo", random() + 1)
|
||
|
|
mlflow.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!")
|
||
|
|
|
||
|
|
mlflow.log_artifacts("outputs")
|