Added normalization of the input values

jakobdanel · Jan 19, 2024 · afd3dc3 · afd3dc3
1 parent e3626a3
commit afd3dc3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/kld.R b/R/kld.R
@@ -23,6 +23,8 @@
 #'
 #' @export
 lfa_kld <- function(p, q, epsilon = 1e-10) {
+  p <- p / sum(p)
+  q <- q / sum(q)
   p <- p + epsilon
   q <- q + epsilon
   return(sum(p * log(p / q)))