diff --git a/clustering/clustering_base.py b/clustering/clustering_base.py index 98a260e..2f37b68 100644 --- a/clustering/clustering_base.py +++ b/clustering/clustering_base.py @@ -68,7 +68,7 @@ class clustering_job: def read_distance_mat(self, similarities, use_threads=True): print(similarities) df = pd.read_feather(similarities, use_threads=use_threads) - mat = np.array(df.drop('_subreddit',1)) + mat = np.array(df.drop('_subreddit',axis=1)) n = mat.shape[0] mat[range(n),range(n)] = 1 return (df._subreddit,1-mat)