config.py

import yaml

class Config:
    def __init__(self,projectname:str = "",base_name:str = "") -> None:
        self.yaml_conf = self.load_yaml(projectname=projectname,base_name=base_name)

    def load_yaml(self,projectname:str = "",base_name:str = ""):
        with open(f"model/{projectname}/config.yml", "r", encoding="utf-8") as f:
            conf = yaml.safe_load(f)
        return conf[base_name if base_name else None]
    
    @property
    def validate_step(self) -> int:
        return int(self.yaml_conf['validate_step'])
    
    @property
    def num_epochs(self) -> int:
        return int(self.yaml_conf['num_steps'])
    
    @property
    def batch_size(self) -> int:
        return int(self.yaml_conf['batch_size'])
    
    @property
    def learning_rate(self) -> float:
        return float(self.yaml_conf['learning_rate'])
    
    @property
    def device(self) -> str:
        return str(self.yaml_conf['device'])
    
    @property
    def num_classes(self) -> int:
        return int(self.yaml_conf['num_classes'])
    
    @property
    def sr(self) -> int:
        return int(self.yaml_conf['sr'])
    
    @property
    def max_shift(self) -> int:
        return int(self.yaml_conf['max_shift'])
    
    @property
    def noise_factor(self) -> float:
        return float(self.yaml_conf['noise_factor'])
    
    @property
    def train_countinue(self) -> bool:
        return bool(self.yaml_conf['train_countinue'])
    
    @property
    def log_interval(self) -> int:
        return int(self.yaml_conf['log_interval'])