Select Git revision
show_results.py 1.46 KiB
import sys
import os
import matplotlib.pyplot as plt
import numpy as np
# Ajouter le répertoire parent au chemin de recherche
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
from parsers.jsonParser import parseJsonFile
RESULTS_DIR_NAME = "./results"
RESULTS_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), RESULTS_DIR_NAME))
try:
results = parseJsonFile(f"{RESULTS_DIR}/zero_shot/v2/condensed_results.json")
print(results)
except Exception as e:
print(f"Error: {e}")
data = results
models = [entry["model"] for entry in data]
categories = [entry["Name"] for entry in data[0]["data"]]
fig, axes = plt.subplots(5, 1, figsize=(18, 50), sharey=True)
colors = ["skyblue", "orange", "green"]
metrics = ["TPR", "TNR", "Precision"]
bar_width = 0.2
x_positions = np.arange(len(models))
for i, category in enumerate(categories):
ax = axes[i]
for j, metric in enumerate(metrics):
values = []
for model_data in data:
category_data = next(item for item in model_data["data"] if item["Name"] == category)
values.append(category_data[metric])
ax.bar(x_positions + j * bar_width, values, width=bar_width, label=metric, color=colors[j])
ax.set_title(category)
ax.set_xticks(x_positions + bar_width)
ax.set_xticklabels(models, rotation=15)
ax.set_ylabel("Valeur")
ax.legend()
# plt.tight_layout()
plt.savefig(f"{RESULTS_DIR}/zero_shot/v2/condensed_results.png")