auto-label: less sensitive for some classes (bluesky-social#1071)

rcg4u · May 18, 2023 · d878564 · d878564
1 parent cf36b36
commit d878564
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/pds/src/labeler/hive.ts b/packages/pds/src/labeler/hive.ts
@@ -151,6 +151,8 @@ const labelForClass = {
   very_bloody: 'gore',
   human_corpse: 'corpse',
   hanging: 'corpse',
+}
+const labelForClassLessSensitive = {
   yes_self_harm: 'self-harm',
 }
 
@@ -161,6 +163,11 @@ export const summarizeLabels = (classes: HiveRespClass[]): string[] => {
       labels.push(labelForClass[cls.class])
     }
   }
+  for (const cls of classes) {
+    if (labelForClassLessSensitive[cls.class] && cls.score >= 0.96) {
+      labels.push(labelForClassLessSensitive[cls.class])
+    }
+  }
   return labels
 }