Skip to content
Extraits de code Groupes Projets
configs.py 451 octets
Newer Older
  • Learn to ignore specific revisions
  • Adrien Payen's avatar
    Adrien Payen a validé
    # local imports
    from models import *
    
    
    class EvalConfig:
        
        models = [
            ("baseline_1", ModelBaseline1, {}),  # model_name, model class, model parameters (dict)
        ]
        split_metrics = ["mae"]
        loo_metrics = []
        full_metrics = []
    
        # Split parameters
    
    Adrien Payen's avatar
    Adrien Payen a validé
        test_size = 0.3 #None  # -- configure the test_size (from 0 to 1) --
    
    Adrien Payen's avatar
    Adrien Payen a validé
    
        # Loo parameters
    
    Adrien Payen's avatar
    Adrien Payen a validé
        top_n_value =  10 #None  # -- configure the numer of recommendations (> 1) --