From 131ca6891b9941da42c729168481296eaab4a47e Mon Sep 17 00:00:00 2001 From: Bobby Date: Sun, 4 Dec 2022 17:39:01 -0500 Subject: Fixed Plots --- src/helpers/preprocessor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/helpers/preprocessor.py') diff --git a/src/helpers/preprocessor.py b/src/helpers/preprocessor.py index d358d69..9c58b24 100644 --- a/src/helpers/preprocessor.py +++ b/src/helpers/preprocessor.py @@ -62,7 +62,7 @@ class Preprocessor: self.df['day'] = self.df['created_utc'].dt.day_name() # self.df.drop('created_utc', axis=1, inplace=True) - cols_to_keep = ['title', 'selftext', 'link_flair_text', 'subreddit', 'ups', 'num_comments', 'hour', 'day', 'distinguished', 'author_premium', 'subreddit_subscribers', 'author', 'score', 'created_utc', 'upvote_ratio'] + cols_to_keep = ['title', 'selftext', 'link_flair_text', 'subreddit', 'ups', 'num_comments', 'hour', 'day', 'distinguished', 'author_premium', 'subreddit_subscribers', 'author', 'score', 'created_utc', 'upvote_ratio', 'total_awards_received'] self.df = self.df[cols_to_keep] # "distinguished" coloumn has 2 values - "moderator" and "" - We can convert this to a boolean column -- cgit v1.2.3