Skip to content
Snippets Groups Projects
Select Git revision
  • 46ca719226ebbc9a3124726a073a11b701c5f58c
  • live_exam_os_ubuntu default protected
2 results

vmEdit.go

Blame
  • 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")