collective/cdsc_reddit
13
0
Fork 0
Code

make pass keyword arg to dataframe.drop

Browse Source
This commit is contained in:
Nathan TeBlunthuis 2023-05-31 09:47:21 -07:00
parent 07b0dff9bc
commit 9345f9de94
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clustering/clustering_base.py
View File

@ -68,7 +68,7 @@ class clustering_job:
def read_distance_mat(self, similarities, use_threads=True): def read_distance_mat(self, similarities, use_threads=True):
print(similarities) print(similarities)
df = pd.read_feather(similarities, use_threads=use_threads) 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] n = mat.shape[0]
mat[range(n),range(n)] = 1 mat[range(n),range(n)] = 1
return (df._subreddit,1-mat) return (df._subreddit,1-mat)