Slightly improved language detection from cld3.

desktop-app · Dec 8, 2022 · 68e0b6f · 68e0b6f
1 parent 95d14bb
commit 68e0b6f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/spellcheck/third_party/language_cld3.cpp b/spellcheck/third_party/language_cld3.cpp
@@ -21,11 +21,12 @@ RecognitionResult Recognize(QStringView text) {
 	const auto string = std::string(text.toUtf8().constData());
 	const auto results = lang_id.FindTopNMostFreqLangs(string, kMaxLangs);
 
-	auto maxProbability = 0.;
+	auto maxRatio = 0.;
 	auto final = NNetLanguageIdentifier::Result();
 	for (const auto &result : results) {
-		if (result.probability > maxProbability) {
-			maxProbability = result.probability;
+		const auto ratio = result.probability * result.proportion;
+		if (ratio > maxRatio) {
+			maxRatio = ratio;
 			final = result;
 		}
 	}