From bb7dbca004d06fed1ec0fb2edfd7056fafa983a6 Mon Sep 17 00:00:00 2001
From: Naufal-Nalendra <16521240@std.stei.itb.ac.id>
Date: Thu, 9 Jan 2025 08:52:04 +0700
Subject: [PATCH] fix: model pipeline
---
.gitlab-ci.yml | 1 +
model/train_model.py | 4 ++++
2 files changed, 5 insertions(+)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 29a9ffa..265fa2d 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -19,6 +19,7 @@ train_model:
- pip install numpy pandas scikit-learn mlflow
# Train the model and log to MLflow
+ - mlflow server --backend-store-uri sqlite:///mlflow.db --default-artifact-root ./mlruns --host 127.0.0.1 --port 5000 &
- python model/train_model.py
only:
diff --git a/model/train_model.py b/model/train_model.py
index 69f86cb..9b68415 100644
--- a/model/train_model.py
+++ b/model/train_model.py
@@ -7,6 +7,7 @@ from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score
import mlflow
import mlflow.sklearn
+from mlflow.tracking import MlflowClient
import os
@@ -35,6 +36,9 @@ ensemble_model = VotingClassifier(
voting='soft'
)
+mlflow.set_tracking_uri("http://127.0.0.1:5000")
+mlflow.set_experiment("Customer Churn Model")
+
# Start MLflow run
with mlflow.start_run():
# Train the model
--
GitLab