Fixed an error in p-value computation for TcpClassifier.

donlnz · Jan 29, 2018 · cdef0d1 · cdef0d1
1 parent 2e26f4c
commit cdef0d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nonconformist/cp.py b/nonconformist/cp.py
@@ -134,10 +134,10 @@ def predict(self, x, significance=None):
 				train_y = np.hstack([self.train_y, y])
 				self.base_icp.fit(train_x, train_y)
 				scores = self.base_icp.nc_function.score(train_x, train_y)
-				n_gt = (scores[:-1] > scores[-1]).sum()
-				n_eq = (scores[:-1] == scores[-1]).sum()
+				ngt = (scores[:-1] > scores[-1]).sum()
+				neq = (scores[:-1] == scores[-1]).sum()
 
-				p[i, j] = calc_p(n_train, n_eq, n_gt, self.smoothing)
+				p[i, j] = calc_p(n_train, ngt, neq, self.smoothing)
 
 		if significance is not None:
 			return p > significance