ryssta/PolevalFastAPI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Modify EmotionRecognition evaluation

Browse Source
This commit is contained in:
Ryszard Staruch 2024-01-02 12:22:45 +01:00
parent a7b78f1eaa
commit 99aacdc440
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -82,20 +82,18 @@ def get_emotion_recognition_scores(df_in, df_expected, df_predition):
df_prediction_sentence = df_predition[~text_annotation]
f1_text_score = f1_score(
df_prediction_text.values.flatten(),
df_expected_text.values.flatten(),
df_prediction_text,
df_expected_text,
average='macro',
zero_division=0.0,
labels=[True, False]
zero_division=0.0
)
f1_text_score = f1_text_score * 100
f1_sentence_score = f1_score(
df_expected_sentence.values.flatten(),
df_prediction_sentence.values.flatten(),
df_expected_sentence,
df_prediction_sentence,
average='macro',
zero_division=0.0,
labels=[True, False]
zero_division=0.0
)
f1_sentence_score = f1_sentence_score * 100