corrige pb

InseeFrLab · Feb 20, 2024 · 4d9f1c7 · 4d9f1c7
1 parent ed90988
commit 4d9f1c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/applications/_ape.qmd b/applications/_ape.qmd
@@ -405,10 +405,10 @@ df_test['prediction'] = predictions['prediction'].astype(str).str.replace(r'__la
 ```
 
 ```{python}
-from sklearn.metrics import accuracy_score, f1_score
+from sklearn.metrics import accuracy_score
 accuracy_score(df_test['nace'],df_test['prediction'])
 
-df['match'] = (df['nace']==df['prediction'])
+df_test['match'] = (df_test['nace']==df_test['prediction'])
 
 # Aggregating by 'nace' to count correct predictions and calculate accuracy
 aggregation = df_test.groupby('nace').agg(correct_predictions=('match', 'sum'),