Spaces:
Runtime error
Runtime error
File size: 5,051 Bytes
42da6ea da68771 42da6ea 8268752 42da6ea da68771 42da6ea f228efb da68771 f228efb eae2854 da68771 eae2854 d3b5bd8 da68771 d3b5bd8 | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 | from datetime import datetime
import os
from networksecurity.constant import training_pipeline
print(training_pipeline.PIPELINE_NAME)
print(training_pipeline.ARTIFACT_DIR)
class TrainingPipelineConfig:
def __init__(self,timestamp=datetime.now()):
timestamp=timestamp.strftime("%m_%d_%Y_%H_M_%S")
self.pipeline_name = training_pipeline.PIPELINE_NAME
self.artifact_name = training_pipeline.ARTIFACT_DIR
self.artifact_dir = os.path.join(self.artifact_name,timestamp)
self.model_dir = os.path.join("final_model")
self.timestamp: str = timestamp
class DataIngestionConfig:
def __init__(self,training_pipeline_config: TrainingPipelineConfig):
self.data_ingestion_dir:str = os.path.join(
training_pipeline_config.artifact_dir,training_pipeline.DATA_INGESTION_DIR_NAME
)
self.feature_store_file_path: str = os.path.join(
self.data_ingestion_dir, training_pipeline.DATA_INGESTION_FEATURE_STORE_DIR, training_pipeline.FILE_NAME
)
self.training_file_path: str = os.path.join(
self.data_ingestion_dir, training_pipeline.DATA_INGESTION_INGESTED_DIR, training_pipeline.TRAIN_FILE_NAME
)
self.testing_file_path: str = os.path.join(
self.data_ingestion_dir, training_pipeline.DATA_INGESTION_INGESTED_DIR, training_pipeline.TEST_FILE_NAME
)
self.train_test_split_ratio: float = training_pipeline.DATA_INGESTION_TRAIN_TEST_SPLIT_RATION
self.collection_name: str = training_pipeline.DATA_INGESTION_COLLECTION_NAME
self.database_name: str = training_pipeline.DATA_INGESTION_DATABASE_NAME
class DataValidationConfig:
def __init__(self, training_pipeline_config: TrainingPipelineConfig):
# Main data validation directory
self.data_validation_dir: str = os.path.join(
training_pipeline_config.artifact_dir,
training_pipeline.DATA_VALIDATION_DIR_NAME
)
# Valid data directory
self.valid_data_dir: str = os.path.join(
self.data_validation_dir,
training_pipeline.DATA_VALIDATION_VALID_DIR
)
# Invalid data directory
self.invalid_data_dir: str = os.path.join(
self.data_validation_dir,
training_pipeline.DATA_VALIDATION_INVALID_DIR
)
# Valid train file path
self.valid_train_file_path: str = os.path.join(
self.valid_data_dir,
training_pipeline.TRAIN_FILE_NAME
)
# Valid test file path
self.valid_test_file_path: str = os.path.join(
self.valid_data_dir,
training_pipeline.TEST_FILE_NAME
)
# Invalid train file path
self.invalid_train_file_path: str = os.path.join(
self.invalid_data_dir,
training_pipeline.TRAIN_FILE_NAME
)
# Invalid test file path
self.invalid_test_file_path: str = os.path.join(
self.invalid_data_dir,
training_pipeline.TEST_FILE_NAME
)
# Drift report file path
self.drift_report_file_path: str = os.path.join(
self.data_validation_dir,
training_pipeline.DATA_VALIDATION_DRIFT_REPORT_DIR,
training_pipeline.DATA_VALIDATION_DRIFT_REPORT_FILE_NAME,
)
class DataTransformationConfig:
def __init__(self, training_pipeline_config: TrainingPipelineConfig):
self.data_transformation_dir: str = os.path.join(
training_pipeline_config.artifact_dir,
training_pipeline.DATA_TRANSFORMATION_DIR_NAME
)
self.transformed_train_file_path: str = os.path.join(
self.data_transformation_dir,
training_pipeline.DATA_TRANSFORMATION_TRANSFORMED_DATA_DIR,
training_pipeline.TRAIN_FILE_NAME.replace("csv", "npy")
)
self.transformed_test_file_path: str = os.path.join(
self.data_transformation_dir,
training_pipeline.DATA_TRANSFORMATION_TRANSFORMED_DATA_DIR,
training_pipeline.TEST_FILE_NAME.replace("csv", "npy")
)
self.transformed_object_file_path: str = os.path.join(
self.data_transformation_dir,
training_pipeline.DATA_TRANSFORMATION_TRANSFORMED_OBJECT_DIR,
training_pipeline.PREPROCESSING_OBJECT_FILE_NAME
)
class ModelTrainerConfig:
def __init__(self, training_pipeline_config: TrainingPipelineConfig):
self.model_trainer_dir: str = os.path.join(
training_pipeline_config.artifact_dir, training_pipeline.MODEL_TRAINER_DIR_NAME
)
self.trained_model_file_path: str = os.path.join(
self.model_trainer_dir, training_pipeline.MODEL_TRAINER_TRAINED_MODEL_DIR,
training_pipeline.MODEL_FILE_NAME
)
self.expected_accuracy: float = training_pipeline.MODEL_TRAINER_EXPECTED_SCORE
self.overfitting_underfitting_threshold = training_pipeline.MODEL_TRAINER_OVER_FITTING_UNDER_FITTING_THRESHOLD |