From 2558433173ce546e7be9ef9d92b53a20f5db802e Mon Sep 17 00:00:00 2001 From: Anurag Chatterjee Date: Sun, 30 Aug 2020 21:45:01 +0800 Subject: [PATCH] Add experiment name Signed-off-by: Anurag Chatterjee --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 26765faf..e31713c6 100644 --- a/train.py +++ b/train.py @@ -44,6 +44,7 @@ def eval_metrics(actual, pred): alpha = float(sys.argv[1]) if len(sys.argv) > 1 else 0.5 l1_ratio = float(sys.argv[2]) if len(sys.argv) > 2 else 0.5 + mlflow.set_experiment("/wine-quality-experiment") with mlflow.start_run(): lr = ElasticNet(alpha=alpha, l1_ratio=l1_ratio, random_state=42) lr.fit(train_x, train_y)