Commit
•
62bffd5
1
Parent(s):
163c84f
Update news_category_similar_news_prediction.py
Browse files
news_category_similar_news_prediction.py
CHANGED
@@ -109,6 +109,7 @@ def add_flags_to_prediction_df(df):
|
|
109 |
|
110 |
df.drop(columns=['similar_news', 'src', 'title', 'description'], inplace=True)
|
111 |
df['y_true_proba'] = df['y_pred_proba']
|
|
|
112 |
return df
|
113 |
|
114 |
|
|
|
109 |
|
110 |
df.drop(columns=['similar_news', 'src', 'title', 'description'], inplace=True)
|
111 |
df['y_true_proba'] = df['y_pred_proba']
|
112 |
+
df['y_true'] = df['y_pred']
|
113 |
return df
|
114 |
|
115 |
|