Skip to content
Snippets Groups Projects
Commit f4a9110e authored by Antonio Andriella's avatar Antonio Andriella
Browse files

Fix bug on correct move counter

parent d8dc7464
No related branches found
No related tags found
No related merge requests found
...@@ -32,6 +32,7 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr ...@@ -32,6 +32,7 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr
max_attempt_counter max_attempt_counter
''' '''
# if then else are necessary to classify the task game state into beg, mid, end # if then else are necessary to classify the task game state into beg, mid, end
if user_action == 1: if user_action == 1:
...@@ -51,7 +52,6 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr ...@@ -51,7 +52,6 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr
attempt_counter = 1 attempt_counter = 1
max_attempt_counter += 1 max_attempt_counter += 1
task_progress_counter +=1 task_progress_counter +=1
correct_move_counter += 1
# TODO call the function to compute the state of the game (beg, mid, end) # TODO call the function to compute the state of the game (beg, mid, end)
...@@ -65,6 +65,7 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr ...@@ -65,6 +65,7 @@ def compute_next_state(user_action, task_progress_counter, attempt_counter, corr
else: else:
game_state_counter = 3 game_state_counter = 3
next_state = (game_state_counter, attempt_counter, user_action) next_state = (game_state_counter, attempt_counter, user_action)
return next_state, task_progress_counter, game_state_counter, attempt_counter, correct_move_counter, wrong_move_counter, timeout_counter, max_attempt_counter return next_state, task_progress_counter, game_state_counter, attempt_counter, correct_move_counter, wrong_move_counter, timeout_counter, max_attempt_counter
...@@ -280,12 +281,14 @@ def simulation(bn_model_user_action, var_user_action_target_action, bn_model_use ...@@ -280,12 +281,14 @@ def simulation(bn_model_user_action, var_user_action_target_action, bn_model_use
print("current_state ", current_state, " next_state ", next_state) print("current_state ", current_state, " next_state ", next_state)
####################################END of EPISODE####################################### ####################################END of EPISODE#######################################
print("task_evolution {}, attempt_counter {}, correct_counter {}, wrong_counter {}, timeout_counter {}".format(game_state_counter, iter_counter, correct_move_counter, wrong_move_counter, timeout_counter)) print("game_state_counter {}, iter_counter {}, correct_counter {}, wrong_counter {}, "
print("robot_assistance_per_action {}".format(robot_assistance_per_action)) "timeout_counter {}, max_attempt {}".format(game_state_counter, iter_counter, correct_move_counter,
print("attempt_counter_per_action {}".format(attempt_counter_per_action)) wrong_move_counter, timeout_counter, max_attempt_counter))
print("game_state_counter_per_action {}".format(game_state_counter_per_action)) # print("robot_assistance_per_action {}".format(robot_assistance_per_action))
print("robot_feedback_per_action {}".format(robot_feedback_per_action)) # print("attempt_counter_per_action {}".format(attempt_counter_per_action))
print("iter {}, correct {}, wrong {}, timeout {}".format(iter_counter, correct_move_counter, wrong_move_counter, timeout_counter)) # print("game_state_counter_per_action {}".format(game_state_counter_per_action))
# print("robot_feedback_per_action {}".format(robot_feedback_per_action))
# print("iter {}, correct {}, wrong {}, timeout {}".format(iter_counter, correct_move_counter, wrong_move_counter, timeout_counter))
#save episode #save episode
episodes.append(episode) episodes.append(episode)
...@@ -349,7 +352,7 @@ def simulation(bn_model_user_action, var_user_action_target_action, bn_model_use ...@@ -349,7 +352,7 @@ def simulation(bn_model_user_action, var_user_action_target_action, bn_model_use
#SIMULATION PARAMS #SIMULATION PARAMS
epochs = 100 epochs = 10
#initialise the robot #initialise the robot
bn_model_robot_assistance = bnlearn.import_DAG('bn_robot_model/robot_assistive_model.bif') bn_model_robot_assistance = bnlearn.import_DAG('bn_robot_model/robot_assistive_model.bif')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment