more baselines, kde, and dm
This commit is contained in:
parent
6b754dd845
commit
ea1e2d2813
|
@ -37,12 +37,18 @@ def wrap_params(cls_params:dict, prefix:str):
|
|||
def baselines():
|
||||
|
||||
q_params = wrap_params(lr_params, 'classifier')
|
||||
kde_params = {**q_params, 'bandwidth': np.linspace(0.01, 0.20, 20)}
|
||||
dm_params = {**q_params, 'nbins': [2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 64]}
|
||||
|
||||
yield CC(new_cls()), "CC", q_params
|
||||
yield ACC(new_cls()), "ACC", q_params
|
||||
yield PCC(new_cls()), "PCC", q_params
|
||||
yield PACC(new_cls()), "PACC", q_params
|
||||
yield KDEyML(new_cls()), "KDEy-ML", {**q_params, 'bandwidth': np.linspace(0.01, 0.20, 20)}
|
||||
yield SLD(new_cls()), "SLD", q_params
|
||||
yield KDEyML(new_cls()), "KDEy-ML", kde_params
|
||||
yield KDEyHD(new_cls()), "KDEy-HD", kde_params
|
||||
# yield KDEyCS(new_cls()), "KDEy-CS", kde_params
|
||||
yield DMy(new_cls()), "DMy", dm_params
|
||||
|
||||
|
||||
def main(args):
|
||||
|
|
|
@ -25,7 +25,7 @@ fi
|
|||
|
||||
for task in T1 T2 T3 T4 ; do
|
||||
|
||||
# PYTHONPATH=.:scripts/:.. python3 baselines.py $task data/
|
||||
PYTHONPATH=.:scripts/:.. python3 baselines.py $task data/
|
||||
|
||||
TEST_SAMPLES=data/lequa2024/$task/public/test_samples
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
from collections import defaultdict
|
||||
import matplotlib.pyplot as plt
|
||||
from matplotlib.cm import get_cmap
|
||||
from matplotlib.pyplot import get_cmap
|
||||
import numpy as np
|
||||
from matplotlib import cm
|
||||
from scipy.stats import ttest_ind_from_stats
|
||||
|
|
Loading…
Reference in New Issue