fix: confidence algorithm

JacobLinCool · Jan 18, 2024 · f5f0ead · f5f0ead
1 parent 1c723a8
commit f5f0ead
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/.changeset/quiet-tables-walk.md b/.changeset/quiet-tables-walk.md
@@ -0,0 +1,5 @@
+---
+"smart-whisper": patch
+---
+
+Fix confidence algorithm
diff --git a/src/binding/transcribe.cc b/src/binding/transcribe.cc
@@ -253,9 +253,9 @@ class TranscribeWorker : public Napi::AsyncProgressQueueWorker<int> {
                     max_p = std::max(max_p, token.p);
                 }
 
-                if (tokens > 2) {
-                    confidence = (confidence - min_p - max_p) / (tokens - 2 - skips);
-                } else {
+                if (tokens - skips > 2) {
+                    confidence = (confidence - min_p - max_p) / (tokens - skips - 2);
+                } else if (tokens - skips > 0) {
                     confidence = confidence / (tokens - skips);
                 }