Training Pipeline Update

KalyanM45 · Jan 2, 2024 · 11ffdea · 11ffdea
1 parent 1e955c0
commit 11ffdea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/Multi_Disease_System/Heart_Disease_Prediction/pipelines/Training_pipeline.py b/src/Multi_Disease_System/Heart_Disease_Prediction/pipelines/Training_pipeline.py
@@ -1,6 +1,6 @@
-from src.Heart_Disease_Prediction.components.Data_ingestion import DataIngestion
-from src.Heart_Disease_Prediction.components.Data_transformation import DataTransformation
-from src.Heart_Disease_Prediction.components.Model_trainer import ModelTrainer
+from src.Multi_Disease_System.Heart_Disease_Prediction.components.Data_ingestion import DataIngestion
+from src.Multi_Disease_System.Heart_Disease_Prediction.components.Data_transformation import DataTransformation
+from src.Multi_Disease_System.Heart_Disease_Prediction.components.Model_trainer import ModelTrainer
 
 class Heart_Disease:
 

diff --git a/src/Multi_Disease_System/Training_Pipelines/Training_Pipelines.py b/src/Multi_Disease_System/Training_Pipelines/Training_Pipelines.py
@@ -1,3 +1,3 @@
-from src.Heart_Disease_Prediction.pipelines.Training_pipeline import Heart_Disease
+from src.Multi_Disease_System.Heart_Disease_Prediction.pipelines.Training_pipeline import Heart_Disease
 heart = Heart_Disease()
 heart.train_heart()