restoring some commented code

This commit is contained in:
Alejandro Moreo Fernandez 2024-02-14 18:52:11 +01:00
parent 323749178b
commit 451c938171
1 changed files with 8 additions and 10 deletions

View File

@ -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):