OxbridgeEconomics
commited on
Commit
·
2dbc5e6
1
Parent(s):
c693abc
commit
Browse files
controllers/vectorizer.py
CHANGED
|
@@ -149,6 +149,7 @@ def vectorize(article):
|
|
| 149 |
"""
|
| 150 |
df = pd.DataFrame(article)
|
| 151 |
df = df[['id','site','title','titleCN','contentCN','category','author','content','subtitle','publishDate','link','attachment','sentimentScore','sentimentLabel']]
|
|
|
|
| 152 |
docs = split_documents(df)
|
| 153 |
documents_list = documents_to_list_of_dicts(docs)
|
| 154 |
for doc in documents_list:
|
|
|
|
| 149 |
"""
|
| 150 |
df = pd.DataFrame(article)
|
| 151 |
df = df[['id','site','title','titleCN','contentCN','category','author','content','subtitle','publishDate','link','attachment','sentimentScore','sentimentLabel']]
|
| 152 |
+
df['sentimentScore'] = df['sentimentScore'].round(2)
|
| 153 |
docs = split_documents(df)
|
| 154 |
documents_list = documents_to_list_of_dicts(docs)
|
| 155 |
for doc in documents_list:
|