groceryheist/cdsc_reddit
1
0
Fork 0
Code

bugfix

Browse Source
This commit is contained in:
Nathan TeBlunthuis 2024-12-28 20:18:53 -08:00
parent d9db21686d
commit a9b296dd73
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
similarities/similarities_helper.py
View File

@ -16,6 +16,7 @@ import pathlib
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
import pickle import pickle
import pytz
class tf_weight(Enum): class tf_weight(Enum):
MaxTF = 1 MaxTF = 1
@ -186,7 +187,7 @@ def write_weekly_similarities(path, sims, week, names):
# reformat as a pairwise list # reformat as a pairwise list
sims = sims.melt(id_vars=['_subreddit','week'],value_vars=names.subreddit.values) sims = sims.melt(id_vars=['_subreddit','week'],value_vars=names.subreddit.values)
sims.to_parquet(p / week) sims.to_parquet(p / week.item().replace(tzinfo=pytz.UTC).isoformat('T'))
def column_overlaps(mat): def column_overlaps(mat):
non_zeros = (mat != 0).astype('double') non_zeros = (mat != 0).astype('double')