fix cosine bug

805094591 · Jan 9, 2019 · 57b9385 · 57b9385
1 parent 97898f6
commit 57b9385
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/utils.cpp b/src/common/utils.cpp
@@ -329,7 +329,7 @@ float cosine_similarity(std::vector<analysis_token_t>& src_tokens, std::vector<a
 
     while (analysis_it != src_tokens.end() && retrieval_it != trg_tokens.end()) {
         if (analysis_it->buffer == retrieval_it->buffer) {
-            product = analysis_it->analysis_term_weight * retrieval_it->analysis_term_weight;
+            product += analysis_it->analysis_term_weight * retrieval_it->analysis_term_weight;
             norm_analysis += analysis_it->analysis_term_weight * analysis_it->analysis_term_weight;
             norm_retrieval += retrieval_it->analysis_term_weight * retrieval_it->analysis_term_weight;
             ++analysis_it;