File size: 970 Bytes
ab80e91 |
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 |
"""
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
|