From 451c93817155567b3f3884f5a9f982f96ff4e6a5 Mon Sep 17 00:00:00 2001 From: Alejandro Moreo Date: Wed, 14 Feb 2024 18:52:11 +0100 Subject: [PATCH] restoring some commented code --- LeQua2024/baselines.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/LeQua2024/baselines.py b/LeQua2024/baselines.py index c285199..28a19f0 100644 --- a/LeQua2024/baselines.py +++ b/LeQua2024/baselines.py @@ -35,18 +35,16 @@ def baselines(): q_params = wrap_params(lr_params, 'classifier') - # 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 EMQ(CalibratedClassifierCV(new_cls())), "SLD-Platt", wrap_params(wrap_params(lr_params, 'estimator'), 'classifier') - # yield EMQ(new_cls()), "SLD", q_params + 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 EMQ(CalibratedClassifierCV(new_cls())), "SLD-Platt", wrap_params(wrap_params(lr_params, 'estimator'), 'classifier') + yield EMQ(new_cls()), "SLD", q_params # yield EMQ(new_cls()), "SLD-BCTS", {**q_params, 'recalib': ['bcts'], 'val_split': [5]} yield MLPE(), "MLPE", None - # if args.task in BINARY_TASKS: - # yield MS2(new_cls()), "MedianSweep2", q_params - # yield KDEyML(new_cls()), "KDEy-ML" - # yield MLPE(), "MLPE" + if args.task in BINARY_TASKS: + yield MS2(new_cls()), "MedianSweep2", q_params def main(args):