Commit
•
1e46c79
1
Parent(s):
f9f964a
Update news_category_similar_news_prediction.py
Browse files
news_category_similar_news_prediction.py
CHANGED
@@ -157,7 +157,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
|
|
157 |
|
158 |
headlines = [*new_news['title']].copy()
|
159 |
|
160 |
-
headlines_desc = [*
|
161 |
label, prob = inference(headlines_desc, interpreter, label_encoder, tokenizer)
|
162 |
|
163 |
|
|
|
157 |
|
158 |
headlines = [*new_news['title']].copy()
|
159 |
|
160 |
+
headlines_desc = [*new_news['title'] + ". " + new_news['description']].copy()
|
161 |
label, prob = inference(headlines_desc, interpreter, label_encoder, tokenizer)
|
162 |
|
163 |
|