remove models

kachiann · Aug 19, 2024 · bb7b0f0 · bb7b0f0
1 parent 27683c7
commit bb7b0f0
Show file tree

Hide file tree

Showing 19 changed files with 119 additions and 104,288 deletions.
diff --git a/models/DecisionTreeRegressor.pkl b/models/DecisionTreeRegressor.pkl
diff --git a/models/DecisionTreeRegressor_model.pkl b/models/DecisionTreeRegressor_model.pkl
diff --git a/models/DecisionTreeRegressor_x_test.csv b/models/DecisionTreeRegressor_x_test.csv
diff --git a/models/DecisionTreeRegressor_x_train.csv b/models/DecisionTreeRegressor_x_train.csv
diff --git a/models/LinearRegression.pkl b/models/LinearRegression.pkl
diff --git a/models/LinearRegression_x_test.csv b/models/LinearRegression_x_test.csv
diff --git a/models/LinearRegression_x_train.csv b/models/LinearRegression_x_train.csv
diff --git a/models/x_test.csv b/models/x_test.csv
diff --git a/models/x_train.csv b/models/x_train.csv
diff --git a/src/experiment_tracking.py b/src/experiment_tracking.py
@@ -54,11 +54,16 @@ def train_and_log_model(
     Returns:
         run_id: The ID of the MLflow run.
     """
+    # Ensure the models directory exists
+    models_dir = os.path.join(os.getcwd(), "models")
+    os.makedirs(models_dir, exist_ok=True)
+
     with mlflow.start_run(run_name=model_name) as run:
         mlflow.set_tag("model", model_name)
 
-        # Log the full dataset as an artifact
-        mlflow.log_artifact(dataset_path, artifact_path="data")
+        if dataset_path:
+            # Log the full dataset as an artifact
+            mlflow.log_artifact(dataset_path, artifact_path="data")
 
         # Save and log the training dataset as a CSV file
         train_csv_path = os.path.join(models_dir, f"{model_name}_x_train.csv")
@@ -103,6 +108,7 @@ def train_and_log_model(
         return run.info.run_id
 
 
+
 def main():
     """Main function to load data, train models, and log to MLflow."""
     # Path to the dataset

diff --git a/src/models/DecisionTreeRegressor.pkl b/src/models/DecisionTreeRegressor.pkl