groceryheist/cdsc_reddit
1
0
Fork 0
Code

bugfix in completing tfidf similarity matrices.

Browse Source
This commit is contained in:
Nate E TeBlunthuis 2020-11-12 11:47:53 -08:00
parent 220a540beb
commit e794214653
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
author_cosine_similarity.py
View File

@ -71,8 +71,8 @@ https://stanford.edu/~rezab/papers/dimsum.pdf. If similarity_threshold=0 we get
similarities = similarities.join(df, on='j') similarities = similarities.join(df, on='j')
similarities = similarities.rename(columns={'subreddit':"subreddit_j"}) similarities = similarities.rename(columns={'subreddit':"subreddit_j"})
similarities.write_feather(output_feather) similarities.to_feather(output_feather)
similarities.write_csv(output_csv) similarities.to_csv(output_csv)
return similarities return similarities
if __name__ == '__main__': if __name__ == '__main__':

4
term_cosine_similarity.py
View File

@ -71,8 +71,8 @@ https://stanford.edu/~rezab/papers/dimsum.pdf. If similarity_threshold=0 we get
similarities = similarities.join(df, on='j') similarities = similarities.join(df, on='j')
similarities = similarities.rename(columns={'subreddit':"subreddit_j"}) similarities = similarities.rename(columns={'subreddit':"subreddit_j"})
similarities.write_feather(output_feather) similarities.to_feather(output_feather)
similarities.write_csv(output_csv) similarities.to_csv(output_csv)
return similarities return similarities
if __name__ == '__main__': if __name__ == '__main__':