diff --git a/pyvbmc_optimization_noexpit.py b/pyvbmc_optimization_noexpit.py index 6b54bc5..5d1b765 100644 --- a/pyvbmc_optimization_noexpit.py +++ b/pyvbmc_optimization_noexpit.py @@ -551,6 +551,7 @@ def fit_all_participants( method: str = "VBMC", n_participants: Optional[int] = None, verbose: bool = True, + plot: bool = True, ) -> Dict[str, List[Dict]]: """ Ajuste tous les modèles pour tous les participants. @@ -591,7 +592,7 @@ def fit_all_participants( try: if method == "VBMC": result = fit_participant_pyvbmc( - participant_data, model_config, verbose=False + participant_data, model_config, verbose=False, plot=plot ) else: result = fit_participant_deoptim( @@ -877,6 +878,7 @@ if __name__ == "__main__": method=method, n_participants=1, # Set to a number to limit for testing verbose=True, + plot=True, ) # Comparaison des modèles