cpm.datasets#

Example datasets that ship with cpm. They are used throughout the tutorials and examples.

load_bandit_data

Load the 2-arm bandit dataset from a CSV file.

load_two_step_data

Load the adult two-step task dataset of Smid et al. (2022).

load_model_based_model_free

Load the two-step task dataset from BrainExplorer.

load_blocking_data

Load the test-stage ratings of the blocking experiment of Spicer et al. (2021).

load_risky_choices

Load the risky choices dataset from a CSV file.

load_metacognition_data

Load the metacognition dataset from a CSV file.

load_csv

Load a CSV file from the data module.