From a8a92d30df40badc028b7253d1113938e8f00b2e Mon Sep 17 00:00:00 2001 From: Nathan TeBlunthuis Date: Thu, 19 Dec 2024 23:34:55 -0800 Subject: [PATCH] bugfix --- similarities/tfidf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/similarities/tfidf.py b/similarities/tfidf.py index 5fae483..1f51793 100755 --- a/similarities/tfidf.py +++ b/similarities/tfidf.py @@ -7,6 +7,7 @@ from functools import partial def _tfidf_wrapper(func, inpath, outpath, topN, term_colname, exclude, included_subreddits, included_terms=None, min_df=None, max_df=None): #spark = SparkSession.builder.config(map={'spark.executor.memory':'900g','spark.executor.cores':128}).getOrCreate() + spark = SparkSession.builder.getOrCreate() df = spark.read.parquet(inpath) df = df.repartition(128*15, ['subreddit',term_colname]) df = df.filter(~ f.col(term_colname).isin(exclude))