def save_agent_trials_results(selected_options, visited_objects,
gained_rewards, predictions, reaction_times):
TrialsInfo.set_subjects_selection(selected_options)
TrialsInfo.set_visited_objects(visited_objects)
TrialsInfo.set_gained_rewards(gained_rewards)
TrialsInfo.set_subject_predictions(predictions)
TrialsInfo.set_selection_reaction_times(reaction_times)
def save_trials_results(selected_keys, reaction_times, predictions=None):
selected_options, visited_objects, rewards = [], [], []
for block in range(TaskParams.num_of_blocks):
block_selected_option, block_visited_objects, block_rewards = \
TaskLogics.block_trials_results(
selected_keys[block],
TrialsInfo.trials_pairs[block],
TrialsInfo.trials_availables_objects[block],
TrialsInfo.available_objects_actual_rewards[block]
)
selected_options.append(block_selected_option)
visited_objects.append(block_visited_objects)
rewards.append(block_rewards)
TrialsInfo.set_subjects_selection(selected_options)
TrialsInfo.set_visited_objects(visited_objects)
TrialsInfo.set_gained_rewards(rewards)
TrialsInfo.set_selection_reaction_times(reaction_times)