diff --git a/freeforestml/model.py b/freeforestml/model.py index f22aa51c86d9a29c89ea921fbbfa3f5fb12faa19..3ef7179a308b8c48243ca80c78a7b408133ca766 100644 --- a/freeforestml/model.py +++ b/freeforestml/model.py @@ -5,6 +5,7 @@ import sys import h5py import json import socket +import warnings import multiprocessing as mp import numpy as np @@ -1071,8 +1072,8 @@ class HepNetSearch: self.tuner_settings = None self.ETH_IP = ETH_IP - if tuner_name not in ['Hyperband', 'RandomSearch', 'BayesianOptimization']: - raise ValueError("%s is an invalid tuner. It must be either Hyperband, etc.." % repr(tuner)) + if tuner_name not in ['Hyperband', 'RandomSearch', 'BayesianOptimization', 'GridSearch']: + warnings.warn("%s is not a tested tuner, the program might break.\nTested tuners: " % (tuner_name,repr(tuner)) ) else: self.tuner_name = tuner_name diff --git a/freeforestml/plot.py b/freeforestml/plot.py index 3eda8b4b8ed6fbbfcbb5a053dbf329eef346f08a..21f52fffd7f30ac1fb8827e1ebd15adfb1dd23bc 100644 --- a/freeforestml/plot.py +++ b/freeforestml/plot.py @@ -507,7 +507,7 @@ def roc(df, signal_process, background_process, discriminant, steps=100, data = pd.DataFrame({"Signal efficiency": signal_effs, "1 - Background efficiency": background_ieffs}) sns.lineplot(x="Signal efficiency", y="1 - Background efficiency", data=data, - ax=axes, ci=None, label=discriminant.name) + ax=axes, errorbar=None, label=discriminant.name) axes.plot([0, 1], [1, 0], color='gray', linestyle=':') axes.set_xlim((0, 1))