summaryrefslogtreecommitdiffstats
path: root/experiments/ml.pyx
diff options
context:
space:
mode:
authorBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-11 21:27:14 -0400
committerBen Green <ben@SEASITs-MacBook-Pro.local>2015-06-11 21:27:14 -0400
commit1143cf60812c37957c81d39d6180921460e62901 (patch)
tree5231b7fe4f3e27d00072d652297208660e47bb5a /experiments/ml.pyx
parent18aa2dd933f4c76ebd374b42fa10d0067870f135 (diff)
downloadcriminal_cascades-1143cf60812c37957c81d39d6180921460e62901.tar.gz
started adding weights into edge probabilities
Diffstat (limited to 'experiments/ml.pyx')
-rw-r--r--experiments/ml.pyx2
1 files changed, 2 insertions, 0 deletions
diff --git a/experiments/ml.pyx b/experiments/ml.pyx
index c1a5616..9c570e9 100644
--- a/experiments/ml.pyx
+++ b/experiments/ml.pyx
@@ -6,6 +6,8 @@ from libc.math cimport log, exp
DTYPE = np.float64
ctypedef np.float_t DTYPE_t
+cdef DTYPE_t plogis(DTYPE_t weight, DTYPE_t delta):
+ return 1/(1 + exp(-weight/delta))
cdef DTYPE_t weight_success(int dist, int dt, DTYPE_t alpha,
DTYPE_t delta, DTYPE_t gamma):