X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/blobdiff_plain/5632a971c633834720b1beb7f65f8a5d0924c0e7..HEAD:/tfidf_authors.py diff --git a/tfidf_authors.py b/tfidf_authors.py deleted file mode 100644 index f06a8ce..0000000 --- a/tfidf_authors.py +++ /dev/null @@ -1,19 +0,0 @@ -from pyspark.sql import SparkSession -from similarities_helper import build_tfidf_dataset - -## TODO:need to exclude automoderator / bot posts. -## TODO:need to exclude better handle hyperlinks. -spark = SparkSession.builder.getOrCreate() - -df = spark.read.parquet("/gscratch/comdata/users/nathante/reddit_tfidf_test_authors.parquet_temp/part-00000-d61007de-9cbe-4970-857f-b9fd4b35b741-c000.snappy.parquet") - -include_subs = set(open("/gscratch/comdata/users/nathante/cdsc-reddit/top_25000_subs_by_comments.txt")) -include_subs = {s.strip('\n') for s in include_subs} -df = df.filter(df.author != '[deleted]') -df = df.filter(df.author != 'AutoModerator') - -df = build_tfidf_dataset(df, include_subs, 'author') - -df.cache() - -df.write.parquet('/gscratch/comdata/users/nathante/subreddit_tfidf_authors.parquet',mode='overwrite',compression='snappy')