Commit
•
f97e412
1
Parent(s):
0cb5606
Update news_category_prediction.py
Browse files
news_category_prediction.py
CHANGED
@@ -57,6 +57,8 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
|
|
57 |
old_news = old_news.copy()
|
58 |
logger.warning(f'old news columns: {[*old_news.columns]}')
|
59 |
logger.warning(f'{len(old_news)} old news items found')
|
|
|
|
|
60 |
else:
|
61 |
logger.warning('No old news is found')
|
62 |
old_news = new_news.copy()
|
|
|
57 |
old_news = old_news.copy()
|
58 |
logger.warning(f'old news columns: {[*old_news.columns]}')
|
59 |
logger.warning(f'{len(old_news)} old news items found')
|
60 |
+
old_news.drop(columns='_id', inplace=True)
|
61 |
+
logger.warning('Dropped _id column from old news data frame.')
|
62 |
else:
|
63 |
logger.warning('No old news is found')
|
64 |
old_news = new_news.copy()
|