Convert exponent to float64

adtech-labs · Feb 12, 2017 · e9d82a6 · e9d82a6
1 parent 54545f8
commit e9d82a6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/bayes_logistic/bayes_logistic.py b/bayes_logistic/bayes_logistic.py
@@ -60,8 +60,12 @@ def logistic_prob(X, w):
     # truncate to avoid numerical over/underflow
     z = np.clip(z, -trunc, trunc)
 
+    z = [np.float64(x) for x in z]
+    z = np.array(z)
+
     # calculate logitstic probability
     pr = np.exp(z)
+
     pr = pr / (1. + pr)
 
     return pr
@@ -536,6 +540,9 @@ def bayes_logistic_prob(X, w, H):
     # do a truncation to prevent exp overflow
     z = np.clip(z, -trunc, trunc)
 
+    z = [np.float64(x) for x in z]
+    z = np.array(z)
+
     # get the moderated logistic probability
     pr = np.exp(z)
     pr = pr / (1. + pr)