From faeb1fd7b3c1985a9ca8a93548ec66edc0b1ec80 Mon Sep 17 00:00:00 2001 From: Luna Date: Sat, 10 Jun 2023 18:36:12 -0300 Subject: [PATCH] add practical error column --- main.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 83e5abe..3a43999 100644 --- a/main.py +++ b/main.py @@ -472,7 +472,12 @@ def plot2(output_path, normalized_scores, model_scores): def plot3(output_path, normalized_scores, model_scores): - data_for_df = {"model": [], "errors": [], "rating_errors": []} + data_for_df = { + "model": [], + "errors": [], + "rating_errors": [], + "practical_errors": [], + } for model in sorted( normalized_scores.keys(), @@ -488,9 +493,11 @@ def plot3(output_path, normalized_scores, model_scores): for rating in ["general", "sensitive", "questionable", "explicit"] if rating in score_data["incorrect_tags"] ) + practical_absolute_error = total_incorrect_tags - total_rating_errors data_for_df["errors"].append(total_incorrect_tags) data_for_df["rating_errors"].append(total_rating_errors) + data_for_df["practical_errors"].append(practical_absolute_error) data_for_df["model"].append(model) df = pd.DataFrame(data_for_df) @@ -499,6 +506,7 @@ def plot3(output_path, normalized_scores, model_scores): data=[ go.Bar(name="incorrect tags", x=df.model, y=df.errors), go.Bar(name="incorrect ratings", x=df.model, y=df.rating_errors), + go.Bar(name="practical error", x=df.model, y=df.practical_errors), ] ) pio.write_image(fig, output_path, width=1024, height=800)