Browse Source

Add reduce_user to IntentDetectionSlotFillingRepository

pull/1764/head
Hironsan 2 years ago
parent
commit
03198dfd90
1 changed files with 8 additions and 0 deletions
  1. 8
      backend/data_export/pipeline/repositories.py

8
backend/data_export/pipeline/repositories.py

@ -205,3 +205,11 @@ class IntentDetectionSlotFillingRepository(TextRepository):
label.setdefault("cats", [])
label.setdefault("entities", [])
return label_per_user
def reduce_user(self, label_per_user: Dict[str, Any]):
cats = []
entities = []
for user, label in label_per_user.items():
cats.extend(label.get("cats", []))
entities.extend(label.get("entities", []))
return {"all": {"entities": entities, "cats": cats}}
Loading…
Cancel
Save