INNER CODE UNIT · Python

load_config

datawhalechina/fun-rec · src/funrec/config/__init__.py:127

def load_config(model_name: str) -> Config:
    """
    根据模型名称加载配置，返回Config对象
    
    Args:
        model_name: 模型名称 (如 "fm", "afm")
    
    Returns:
        Config对象，支持config.data、config.features等属性访问
    
    Raises:
        KeyError: 如果模型配置不存在
    """
    config_dict = get_config(model_name)
    
    return Config(
        data=config_dict.get("data", {}),
        features=config_dict.get("features", {}),

View source record →