polarolouis/REE-RL-Lola
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adding plot to no expit version

Browse source
This commit is contained in:
Louis 2025-12-11 10:32:55 +01:00
parent ac97c2e05d
commit bb144010d5
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pyvbmc_optimization_noexpit.py
View file

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