config/config.yaml

artifacts_root: artifacts

data_ingestion:
  root_dir: artifacts/data_ingestion
  source_url: kaggle://playground-series-s6e2
  local_data_file: artifacts/data_ingestion/train_raw.csv
  unzip_dir: artifacts/data_ingestion

data_validation:
  root_dir: artifacts/data_validation
  data_dir: artifacts/data_ingestion/train_raw.csv
  status_file: artifacts/data_validation/status.txt
  schema_file: config/schema.yaml

data_transformation:
  root_dir: artifacts/data_transformation
  data_path: artifacts/data_ingestion/train_raw.csv
  train_path: artifacts/data_transformation/train.csv
  test_path: artifacts/data_transformation/test.csv
  test_size: 0.2
  random_state: 42

feature_engineering:
  root_dir: artifacts/feature_engineering
  train_path: artifacts/data_transformation/train.csv
  test_path: artifacts/data_transformation/test.csv
  output_train_path: artifacts/feature_engineering/train_features.csv
  output_test_path: artifacts/feature_engineering/test_features.csv

model_trainer:
  root_dir: artifacts/model_trainer
  train_data_path: artifacts/feature_engineering/train_features.csv
  test_data_path: artifacts/feature_engineering/test_features.csv
  model_path: artifacts/model_trainer/model
  target_column: Heart Disease

model_evaluation:
  root_dir: artifacts/model_evaluation
  model_path: artifacts/model_trainer/model
  test_data_path: artifacts/feature_engineering/test_features.csv
  metrics_file: artifacts/model_evaluation/metrics.json
  target_column: Heart Disease

model_pusher:
  root_dir: artifacts/model_pusher
  model_path: artifacts/model_trainer/model
  model_registry_path: models/production

mlflow:
  tracking_uri: https://dagshub.com/abheshith7/AutoML-MLOps-PipeLine.mlflow/
  experiment_name: automl_experiment
  run_name: null
  registry_uri: null