Skip to content

Commit 392a38b

Browse files
committed
WIP add action to log an mlflow run
1 parent 7861026 commit 392a38b

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
from django.conf import settings
2+
import djclick as click
3+
from mlflow import MlflowClient
4+
from sklearn.datasets import load_diabetes
5+
from sklearn.ensemble import RandomForestRegressor
6+
from sklearn.model_selection import train_test_split
7+
8+
9+
@click.command()
10+
def command():
11+
click.echo("Running Mlflow experiment")
12+
13+
client = MlflowClient(tracking_uri=settings.MLFLOW_ENDPOINT)

0 commit comments

Comments
 (0)