|
|
""" |
|
|
Configuration classes for the Universal Image Classifier |
|
|
""" |
|
|
from dataclasses import dataclass |
|
|
from typing import Optional, Tuple |
|
|
|
|
|
@dataclass |
|
|
class ModelConfig: |
|
|
"""Configuration for model architecture""" |
|
|
input_height: int = 64 |
|
|
input_width: int = 64 |
|
|
num_classes: int = 10 |
|
|
hidden_dim: int = 128 |
|
|
num_layers: int = 4 |
|
|
dropout_rate: float = 0.1 |
|
|
use_batch_norm: bool = True |
|
|
|
|
|
@dataclass |
|
|
class TrainingConfig: |
|
|
"""Configuration for training parameters""" |
|
|
batch_size: int = 32 |
|
|
learning_rate: float = 0.001 |
|
|
num_epochs: int = 100 |
|
|
weight_decay: float = 1e-4 |
|
|
early_stopping_patience: int = 10 |
|
|
validation_split: float = 0.2 |
|
|
|
|
|
@dataclass |
|
|
class DataConfig: |
|
|
"""Configuration for data processing""" |
|
|
image_size: Tuple[int, int] = (64, 64) |
|
|
normalize_mean: Tuple[float, float, float] = (0.485, 0.456, 0.406) |
|
|
normalize_std: Tuple[float, float, float] = (0.229, 0.224, 0.225) |
|
|
augment_training: bool = True |
|
|
|
