fix n_rounds of bandit_feedback_test

zwcdp · Feb 22, 2021 · 98d2432 · 98d2432
1 parent 5c122f7
commit 98d2432
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/obp/dataset/real.py b/obp/dataset/real.py
@@ -241,7 +241,7 @@ def obtain_batch_bandit_feedback(
                 action_context=self.action_context,
             )
             bandit_feedback_test = dict(
-                n_rounds=n_rounds_train,
+                n_rounds=(self.n_rounds - n_rounds_train),
                 n_actions=self.n_actions,
                 action=self.action[n_rounds_train:],
                 position=self.position[n_rounds_train:],