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)