lalithadevi commited on
Commit
f97e412
1 Parent(s): 0cb5606

Update news_category_prediction.py

Browse files
Files changed (1) hide show
  1. news_category_prediction.py +2 -0
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()