Skip to content
Snippets Groups Projects
Commit 2c11235e authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Tracebacks included

parent 119e2237
No related branches found
No related tags found
No related merge requests found
Pipeline #4216 failed
......@@ -13,7 +13,7 @@ debug: True
add_noise: False
noise_std: 0.0
res_dir: "../results/"
track_tracebacks: False
track_tracebacks: True
# All the classification-realted configuration options
multiclass_method: "oneVersusOne"
......
......@@ -65,7 +65,6 @@ class HPSearch:
self.cv_results_["mean_test_score"] = []
self.cv_results_["params"] = []
n_failed = 0
self.tracebacks = []
self.tracebacks_params = []
for candidate_param_idx, candidate_param in enumerate(self.candidate_params):
test_scores = np.zeros(n_splits) + 1000
......@@ -164,6 +163,7 @@ class Random(RandomizedSearchCV, HPSearch):
self.view_indices = view_indices
self.equivalent_draws = equivalent_draws
self.track_tracebacks = track_tracebacks
self.tracebacks=[]
def get_param_distribs(self, estimator):
if isinstance(estimator, MultiClassWrapper):
......@@ -208,6 +208,7 @@ class Grid(GridSearchCV, HPSearch):
self.available_indices = learning_indices
self.view_indices = view_indices
self.track_tracebacks = track_tracebacks
self.tracebacks = []
def fit(self, X, y=None, groups=None, **fit_params):
if self.framework == "monoview":
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment