diff --git a/LeQua2024/baselines.py b/LeQua2024/baselines.py index d35e09a..0903eb2 100644 --- a/LeQua2024/baselines.py +++ b/LeQua2024/baselines.py @@ -3,6 +3,8 @@ import pickle import os import sys from os.path import join + +import numpy as np from sklearn.linear_model import LogisticRegression as LR from scripts.constants import SAMPLE_SIZE @@ -40,6 +42,7 @@ def baselines(): 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)} def main(args): diff --git a/LeQua2024/run_baselines.sh b/LeQua2024/run_baselines.sh index e3d067c..f87986e 100755 --- a/LeQua2024/run_baselines.sh +++ b/LeQua2024/run_baselines.sh @@ -34,7 +34,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