From 87970d1a5d2ced068cb005ea9ebc18d5d2f41782 Mon Sep 17 00:00:00 2001 From: sheyma Date: Thu, 13 Aug 2015 12:57:05 +0200 Subject: [PATCH] remove flatten method --- mapalign/embed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mapalign/embed.py b/mapalign/embed.py index 59e053d..777c090 100644 --- a/mapalign/embed.py +++ b/mapalign/embed.py @@ -72,7 +72,7 @@ def compute_diffusion_map(L, alpha=0.5, n_components=None, diffusion_time=0, if alpha > 0: # Step 2 - d = np.array(L_alpha.sum(axis=1)).flatten() + d = np.array(L_alpha.sum(axis=1)) d_alpha = np.power(d, -alpha) if use_sparse: L_alpha.data *= d_alpha[L_alpha.indices] @@ -84,7 +84,7 @@ def compute_diffusion_map(L, alpha=0.5, n_components=None, diffusion_time=0, L_alpha = L_alpha * d_alpha[np.newaxis, :] # Step 3 - d_alpha = np.power(np.array(L_alpha.sum(axis=1)).flatten(), -1) + d_alpha = np.power(np.array(L_alpha.sum(axis=1)), -1) if use_sparse: L_alpha.data *= d_alpha[L_alpha.indices] else: