mirror of
https://github.com/Toumash/mlflow-docker
synced 2025-11-04 15:19:21 +01:00
version 0.4
This commit is contained in:
1
.dockerignore
Normal file
1
.dockerignore
Normal file
@@ -0,0 +1 @@
|
|||||||
|
.git
|
||||||
8
Dockerfile
Normal file
8
Dockerfile
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
FROM continuumio/miniconda3:latest
|
||||||
|
|
||||||
|
MAINTAINER JianKai Wang "https://jiankaiwang.no-ip.biz"
|
||||||
|
|
||||||
|
ADD . /app
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN pip install mlflow
|
||||||
20
quickstart/mlflow_tracking.py
Normal file
20
quickstart/mlflow_tracking.py
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
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")
|
||||||
Reference in New Issue
Block a user