diff --git a/src/active_bo_ros/active_bo_ros/interactive_bo.py b/src/active_bo_ros/active_bo_ros/interactive_bo.py index 5547b9f..c587c26 100644 --- a/src/active_bo_ros/active_bo_ros/interactive_bo.py +++ b/src/active_bo_ros/active_bo_ros/interactive_bo.py @@ -143,7 +143,7 @@ class ActiveBOTopic(Node): self.overwrite = msg.overwrite # initialize - self.reward = np.zeros((self.bo_episodes+self.nr_init, self.bo_runs)) + self.reward = np.zeros((self.bo_episodes + self.nr_init - 1, self.bo_runs)) self.best_pol_reward = np.zeros((1, self.bo_runs)) self.best_policy = np.zeros((self.bo_steps, self.bo_runs)) self.best_weights = np.zeros((self.bo_nr_weights, self.bo_runs))