98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
import os.path
|
|
import pickle
|
|
from collections import defaultdict
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
from sklearn.feature_extraction.text import TfidfVectorizer
|
|
from sklearn.linear_model import LogisticRegression
|
|
from sklearn.model_selection import GridSearchCV
|
|
from sklearn.svm import LinearSVC
|
|
|
|
import quapy as qp
|
|
from Retrieval.commons import RetrievedSamples, load_sample
|
|
from Retrieval.experiments import methods, benchmark_name
|
|
from Retrieval.plot_mrae_xaxis_k import load_all_results
|
|
from method.non_aggregative import MaximumLikelihoodPrevalenceEstimation as Naive
|
|
from quapy.method.aggregative import ClassifyAndCount, EMQ, ACC, PCC, PACC, KDEyML
|
|
from quapy.data.base import LabelledCollection
|
|
|
|
from os.path import join
|
|
from tqdm import tqdm
|
|
|
|
from result_table.src.table import Table
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
|
|
data_home = 'data'
|
|
class_mode = 'multiclass'
|
|
|
|
method_names = [name for name, *other in methods(None)]
|
|
|
|
# Ks = [5, 10, 25, 50, 75, 100, 250, 500, 750, 1000]
|
|
Ks = [50, 100, 500, 1000]
|
|
DATA_SIZE = ['10K', '50K', '100K', '500K', '1M', 'FULL']
|
|
CLASS_NAME = ['gender', 'continent', 'years_category']
|
|
all_results = {}
|
|
|
|
|
|
# loads all MRAE results, and returns a dictionary containing the values, which is indexed by:
|
|
# class_name -> data_size -> method_name -> k -> stat -> float
|
|
results = load_all_results()
|
|
|
|
# generates the class-independent, size-independent plots for y-axis=MRAE in which:
|
|
# - the x-axis displays the Ks
|
|
|
|
for class_name in CLASS_NAME:
|
|
for k in Ks:
|
|
|
|
log = True
|
|
|
|
fig, ax = plt.subplots()
|
|
|
|
max_means = []
|
|
for method_name in method_names:
|
|
# class_name -> data_size -> method_name -> k -> stat -> float
|
|
means = [
|
|
results[class_name][data_size][method_name][k]['mean'] for data_size in DATA_SIZE
|
|
]
|
|
stds = [
|
|
results[class_name][data_size][method_name][k]['std'] for data_size in DATA_SIZE
|
|
]
|
|
# max_mean = np.max([
|
|
# results[class_name][data_size][method_name][k]['max'] for data_size in DATA_SIZE
|
|
# ])
|
|
|
|
max_means.append(max(means))
|
|
|
|
style = 'o-' if method_name != 'CC' else '--'
|
|
line = ax.plot(DATA_SIZE, means, style, label=method_name, color=None)
|
|
color = line[-1].get_color()
|
|
if log:
|
|
ax.set_yscale('log')
|
|
# ax.fill_between(Ks, means - stds, means + stds, alpha=0.3, color=color)
|
|
|
|
ax.set_xlabel('training pool size')
|
|
ax.set_ylabel('RAE' + ('(log scale)' if log else ''))
|
|
ax.set_title(f'{class_name} from {k=}')
|
|
ax.set_ylim([0, max(max_means)*1.05])
|
|
|
|
ax.legend()
|
|
|
|
os.makedirs(f'plots/var_size/{class_name}', exist_ok=True)
|
|
plotpath = f'plots/var_size/{class_name}/{k}_mrae.pdf'
|
|
print(f'saving plot in {plotpath}')
|
|
plt.savefig(plotpath, bbox_inches='tight')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|