Weigh the group_hit_counts by EC counts

PROBIC · May 29, 2024 · d7dee7b · d7dee7b
1 parent 59fe08c
commit d7dee7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/Likelihood.hpp b/include/Likelihood.hpp
@@ -117,7 +117,7 @@ class LL_WOR21 : public Likelihood<T> {
 	// Create mask identifying groups that have at least 1 alignment
 	for (size_t i = 0; i < num_ecs; ++i) {
 	    for (size_t j = 0; j < n_groups; ++j) {
-		group_hit_counts[j] += (alignment(j, i) > 0);
+		group_hit_counts[j] += (alignment(j, i) > 0) * alignment.reads_in_ec(i);
 	    }
 	}
 	for (size_t i = 0; i < n_groups; ++i) {