cymr.fit.prepare_lists

cymr.fit.prepare_lists(data, study_keys=None, recall_keys=None, clean=True)

Prepare study and recall data for simulation.

Return data information split by list. This format is similar to frdata structs used in EMBAM.

Parameters:

• data (pandas.DataFrame) – Free recall data in Psifr format.

• study_keys (list of str, optional) – Columns to export for study list data. Default is: [‘input’, ‘item_index’]. Input position is assumed to be one-indexed.

• recall_keys (list of str, optional) – Columns to export for recall list data. Default is: [‘input’, ‘item_index’]. Input position is assumed to be one-indexed.

• clean (bool, optional) – If true, repeats and intrusions will be removed.

Returns:

• study (dict of (str: list of numpy.array)) – Study columns in list format.

• recall (dict of (str: list of numpy.array)) – Recall columns in list format.