Multi_Comparison_Matrix/main.py at main · MSD-IRIMAS/Multi_Comparison_Matrix · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import pandas as pd

from multi_comp_matrix import MCM

if __name__ == "__main__":
    path_res = "./results_example.csv"
    output_dir = "./"

    df_results = pd.read_csv(path_res)

    MCM.compare(
        df_results=df_results,
        output_dir=output_dir,
        pdf_savename="heatmap",
        png_savename="heatmap",
    )

    MCM.compare(
        df_results=df_results,
        output_dir=output_dir,
        excluded_col_comparates=["clf1", "clf3"],
        png_savename="heatline_vertical",
        include_ProbaWinTieLoss=True
    )

    MCM.compare(
        df_results=df_results,
        output_dir=output_dir,
        excluded_row_comparates=["clf1", "clf3"],
        png_savename="heatline_horizontal",
        tex_savename="heatline_horizontal",
    )