| project: | |
| name: "Brain Tumor Classification" | |
| version: "1.0.0" | |
| seed: 42 | |
| data: | |
| train_dir: "./data/Training" | |
| test_dir: "./data/Testing" | |
| image_size: [150, 150] | |
| batch_size: 16 | |
| validation_split: 0.2 | |
| classes: ["glioma", "meningioma", "notumor", "pituitary"] | |
| augmentation: | |
| rotation_range: 15 | |
| width_shift_range: 0.1 | |
| height_shift_range: 0.1 | |
| zoom_range: 0.1 | |
| horizontal_flip: true | |
| brightness_range: [0.9, 1.1] | |
| training: | |
| epochs: 20 | |
| early_stopping_patience: 5 | |
| reduce_lr_patience: 3 | |
| reduce_lr_factor: 0.5 | |
| min_lr: 0.000001 | |
| optuna: | |
| n_trials: 10 | |
| direction: "maximize" | |
| models: | |
| save_dir: "./saved_models" | |
| onnx_dir: "./onnx_models" | |
| logs_dir: "./logs" | |
| mlflow: | |
| dagshub_username: "shoaib.shahriar01" | |
| dagshub_token: "2df3bdfdd248ea99d5af9a5d3e485bfaf6ca7c81" | |
| dagshub_repo: "brain-tumor-classification" | |
| experiment_name: "Brain_Tumor_Classification" | |
| api: | |
| host: "0.0.0.0" | |
| port: 8000 | |
| reload: true | |