kde
This commit is contained in:
parent
8dfb109b41
commit
81fbb54992
|
@ -3,6 +3,8 @@ import pickle
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
from os.path import join
|
from os.path import join
|
||||||
|
|
||||||
|
import numpy as np
|
||||||
from sklearn.linear_model import LogisticRegression as LR
|
from sklearn.linear_model import LogisticRegression as LR
|
||||||
|
|
||||||
from scripts.constants import SAMPLE_SIZE
|
from scripts.constants import SAMPLE_SIZE
|
||||||
|
@ -40,6 +42,7 @@ def baselines():
|
||||||
yield ACC(new_cls()), "ACC", q_params
|
yield ACC(new_cls()), "ACC", q_params
|
||||||
yield PCC(new_cls()), "PCC", q_params
|
yield PCC(new_cls()), "PCC", q_params
|
||||||
yield PACC(new_cls()), "PACC", 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):
|
def main(args):
|
||||||
|
|
|
@ -34,7 +34,7 @@ fi
|
||||||
|
|
||||||
for task in T1 T2 T3 T4 ; do
|
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
|
TEST_SAMPLES=data/lequa2024/$task/public/test_samples
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue