gavinzli commited on
Commit
d69af94
·
1 Parent(s): 579df96

Remove commented-out DataFrame column selection in vectorize function for cleaner code

Browse files
Files changed (1) hide show
  1. controllers/vectorizer.py +2 -2
controllers/vectorizer.py CHANGED
@@ -46,8 +46,8 @@ def vectorize(article):
46
  df = pd.DataFrame(article)
47
  df = df[['id','site','title','titleCN','category','author','content',
48
  'publishDate','link']]
49
- df = df[['id', 'publishdate', 'author', 'category',
50
- 'content', 'referenceid', 'site', 'title', 'link']]
51
  # df['sentimentScore'] = df['sentimentScore'].round(2)
52
  # df['sentimentScore'] = df['sentimentScore'].astype(float)
53
  df['publishDate'] = pd.to_datetime(df['publishDate'])
 
46
  df = pd.DataFrame(article)
47
  df = df[['id','site','title','titleCN','category','author','content',
48
  'publishDate','link']]
49
+ # df = df[['id', 'publishdate', 'author', 'category',
50
+ # 'content', 'referenceid', 'site', 'title', 'link']]
51
  # df['sentimentScore'] = df['sentimentScore'].round(2)
52
  # df['sentimentScore'] = df['sentimentScore'].astype(float)
53
  df['publishDate'] = pd.to_datetime(df['publishDate'])