Merge pull request #7575 from hotzenklotz/conf_mat_response_selector

Amendmend to PR #7423
RasaHQ · Dec 18, 2020 · e59b818 · e59b818
2 parents 7d3e6dd + 043f30d
commit e59b818
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/changelog/7575.bugfix.rst b/changelog/7575.bugfix.rst
@@ -0,0 +1 @@
+Fix scikit-learn crashing during evaluation of `ResponseSelector` predictions.
diff --git a/rasa/nlu/test.py b/rasa/nlu/test.py
@@ -1052,9 +1052,11 @@ def get_eval_data(
 
             response_prediction_full_intent = selector_properties.get(
                 response_prediction_key, {}
-            ).get("full_retrieval_intent", {})
+            ).get("full_retrieval_intent", None)
 
-            response_key = example.get_combined_intent_response_key()
+            response_key = None
+            if example.get(RESPONSE_KEY_ATTRIBUTE) is not None:
+                response_key = example.get_combined_intent_response_key()
 
             response_selection_results.append(
                 ResponseSelectionEvaluationResult(