mirror of
https://github.com/Toumash/mlflow-docker
synced 2025-11-04 23:29:19 +01:00
21 lines
498 B
Python
21 lines
498 B
Python
|
import os
|
||
|
|
from random import random, randint
|
||
|
|
|
||
|
|
from mlflow import log_metric, log_param, log_artifacts
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
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")
|