Fixes #22: final commit to cover randomization tests for softmax

VowpalWabbit · Oct 6, 2014 · e79bdd2 · e79bdd2
1 parent 97ee4b5
commit e79bdd2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/explore/tests/MWTExploreTests.cpp b/explore/tests/MWTExploreTests.cpp
@@ -176,9 +176,9 @@ namespace vw_explore_tests
 			u32 i;
 			for (i = 0; i < num_decisions; i++)
 			{
-				pair<u32, u64> action_and_key = m_mwt->Choose_Action_And_Key(*m_context);
+				u32 action = m_mwt->Choose_Action(*m_context, this->Get_Unique_Key(i + 1));
 				// Action IDs are 1-based
-				actions[action_and_key.first - 1]++;
+				actions[MWTAction::Make_ZeroBased(action)]++;
 			}
 			// Ensure all actions are covered
 			for (i = 0; i < m_num_actions; i++)
@@ -206,9 +206,9 @@ namespace vw_explore_tests
 			u32 i;
 			for (i = 0; i < num_decisions; i++)
 			{
-				pair<u32, u64> action_and_key = m_mwt->Choose_Action_And_Key(*m_context);
+				u32 action = m_mwt->Choose_Action(*m_context, this->Get_Unique_Key(i + 1));
 				// Action IDs are 1-based
-				actions[action_and_key.first - 1]++;
+				actions[MWTAction::Make_ZeroBased(action)]++;
 			}
 			// Ensure all actions are covered
 			for (i = 0; i < m_num_actions; i++)