Adding plot to no expit version

This commit is contained in:
Louis 2025-12-11 10:32:55 +01:00
parent ac97c2e05d
commit bb144010d5

View file

@ -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