Spaces:
Sleeping
Sleeping
File size: 883 Bytes
0928d05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os , sys
from src.Predictive_Maintenance.components.data_ingestion import DataIngestion
from src.Predictive_Maintenance.components.data_transformation import DataTransformation
from src.Predictive_Maintenance.components.model_trainer import ModelTrainer
from src.Predictive_Maintenance.logger import logging
from src.Predictive_Maintenance.exception import CustomException
def main():
try:
obj = DataIngestion()
raw_data_path = obj.initiate_data_ingestion()
data_transformation = DataTransformation()
df_sampled = data_transformation.initiate_data_transformation(raw_data_path)
model_trainer = ModelTrainer()
model_trainer.initiate_model_training(df_sampled)
except Exception as e:
logging.error(f"An error occurred: {e}")
raise CustomException(e, sys)
if __name__ == "__main__":
main() |