fix mlflow

MLFLOW/Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.11
+
+RUN apt-get update && apt-get -y upgrade
+RUN apt-get install -y build-essential
+
+RUN python -m pip install --upgrade pip
+COPY ../requirements.txt /tmp
+RUN python -m pip install -r /tmp/requirements.txt
+
+WORKDIR ./app
+COPY train.py ./
+
+CMD bash

MLFLOW/requirements.txt

@@ -0,0 +1,3 @@
+pandas
+tensorflow
+mlflow

MLFLOW/train.py

@@ -3,8 +3,6 @@ from tensorflow import keras
 from tensorflow.keras import layers
 import argparse
 import mlflow
-
-
 class RegressionModel:
     def __init__(self, optimizer="adam", loss="mean_squared_error"):
         self.model = keras.Sequential([
@@ -48,11 +46,11 @@ parser = argparse.ArgumentParser()
 parser.add_argument('--epochs')
 
 args = parser.parse_args()
-mlflow.set_tracking_uri("http://localhost:5000")
-model = RegressionModel()
 with mlflow.start_run() as run:
+    model = RegressionModel()
+    model.load_data("df_train.csv", "df_test.csv")
     model.train(epochs=int(args.epochs))
+    mlflow.log_param("epochs", int(args.epochs))
     rmse = model.evaluate()
-    mlflow.log_param("epoch", int(args.epochs))
     mlflow.log_metric("rmse", rmse)
 model.save_model()

get_stats.py

@@ -5,9 +5,9 @@ pd.set_option('display.max_columns', None)
 pd.set_option('display.max_rows', None)
 
 
-df = pd.read_csv('./dataset.csv')
-df_train = pd.read_csv('./df_train.csv')
-df_test = pd.read_csv('./df_test.csv')
+df = pd.read_csv('MLFLOW/dataset.csv')
+df_train = pd.read_csv('MLFLOW/df_train.csv')
+df_test = pd.read_csv('MLFLOW/df_test.csv')
 
 
 with open('stats.txt', 'w') as f:

predict.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 
 np.set_printoptions(threshold=np.inf)
 
-data = pd.read_csv("df_test.csv")
+data = pd.read_csv("MLFLOW/df_test.csv")
 X_test = data.drop("Performance Index", axis=1)
 y_test = data["Performance Index"]