fix bug 1784: return probabilities for actions in the correct order (V…

…owpalWabbit#1830)
jackgerrits · May 15, 2019 · fd6b0d0 · fd6b0d0
1 parent 1fe4e6f
commit fd6b0d0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/python/pylibvw.cc b/python/pylibvw.cc
@@ -460,10 +460,16 @@ py::list ex_get_scalars(example_ptr ec)
 py::list ex_get_action_scores(example_ptr ec)
 { py::list values;
   v_array<ACTION_SCORE::action_score> scores = ec->pred.a_s;
+  std::vector<float> ordered_scores(scores.size());
+  for (auto action_score: scores)
+  {
+     ordered_scores[action_score.action] = action_score.score;
+  }
 
-  for (ACTION_SCORE::action_score s : scores)
-  { values.append(s.score);
+  for (auto action_score: ordered_scores)
+  { values.append(action_score);
   }
+
   return values;
 }