Adding plot to no expit version
This commit is contained in:
parent
ac97c2e05d
commit
bb144010d5
1 changed files with 3 additions and 1 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue