diff --git a/mlflow_config.yaml b/mlflow_config.yaml
new file mode 100644
index 0000000..b13701f
--- /dev/null
+++ b/mlflow_config.yaml
@@ -0,0 +1,7 @@
+name: s487187-training
+
+entry_points:
+  main:
+    parameters:
+      epochs: int
+    command: "python train.py --epochs {epochs}"
\ No newline at end of file