diff --git a/train.py b/train.py
index 26765faf..80af3c7c 100644
--- a/train.py
+++ b/train.py
@@ -25,6 +25,7 @@ def eval_metrics(actual, pred):
 
 
 if __name__ == "__main__":
+    mlflow.set_tracking_uri("sqlite:///mlruns.db")
     warnings.filterwarnings("ignore")
     np.random.seed(40)