From 1d701876be8f24d52c785194a7fa996a033be1cf Mon Sep 17 00:00:00 2001 From: Antonio Andriella <aandriella@iri.upc.edu> Date: Thu, 24 Sep 2020 23:42:53 +0200 Subject: [PATCH] replace Robot with Agent --- cognitive_game_vars.py | 4 ++-- episode.py | 10 +++++----- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/cognitive_game_vars.py b/cognitive_game_vars.py index 3571ef6..4fa0f4b 100644 --- a/cognitive_game_vars.py +++ b/cognitive_game_vars.py @@ -43,7 +43,7 @@ class Attention(enum.Enum): name = "attention" counter = 3 -class Robot_Assistance(enum.Enum): +class Agent_Assistance(enum.Enum): lev_0 = 0 lev_1 = 1 lev_2 = 2 @@ -53,7 +53,7 @@ class Robot_Assistance(enum.Enum): name = "robot_assistance" counter = 6 -class Robot_Feedback(enum.Enum): +class Agent_Feedback(enum.Enum): yes = 1 no = 0 name = "robot_feedback" diff --git a/episode.py b/episode.py index cdf63e8..bcc420a 100644 --- a/episode.py +++ b/episode.py @@ -74,8 +74,8 @@ class Episode: ''' It returns the episodes related to the saved file :param file: - :param episode: look at main.py - :param sol_per_pop: look at main.py + :param episode: look at simulation.py + :param sol_per_pop: look at simulation.py :return: a list of episodes ''' print("LOADING...") @@ -167,7 +167,7 @@ class Episode: def main(): pass - # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/episodes.npy" + # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/episodes.npy" # ep = Episode() # episodes = ep.load_episodes(file_path) # initial_state = (1, 1, 0) @@ -187,8 +187,8 @@ def main(): # task_length, n_max_attempt, timeout=0, n_levels_assistance=6) # # # trans_matrix = ep.generate_statistics(env.states, env.action_space, episodes) - # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_occ.npy" - # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_prob.npy" + # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_occ.npy" + # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_prob.npy" # terminal_states = [env.point_to_index(state) for state in final_states] # # -- GitLab