remove np.sqrt and unnecessary commentout

tsurubee · Feb 2, 2019 · 97455e2 · 97455e2
1 parent e08bfb1
commit 97455e2
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/banpei/sst.py b/banpei/sst.py
@@ -134,10 +134,7 @@ def _calculate_score_by_lanczos(self, tra_matrix, test_matrix):
         P = np.dot(tra_matrix, tra_matrix.T)
         T = tridiagonalize_by_lanczos(P, m, k)
         eigenvalue, eigenvectors = tridiag_eigen(T)
-
-        # `eig()` returns unordered eigenvalues,
-        # so the top-r eigenvectors should be picked carefully
-        return 1 - np.sqrt(np.sum(eigenvectors[0, np.argsort(eigenvalue)[::-1][:self.m]] ** 2))
+        return 1 - np.sum(eigenvectors[0, np.argsort(eigenvalue)[::-1][:self.m]] ** 2)
 
     def _extract_matrix(self, data, start, end, w):
         row = w