From b111c2640a3e270a13af6967bb0f35c94c30eb5c Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Wed, 6 Mar 2019 23:04:50 -0500 Subject: bugfix for gamma in kernel --- gensvm/gridsearch.py | 1 + 1 file changed, 1 insertion(+) (limited to 'gensvm') diff --git a/gensvm/gridsearch.py b/gensvm/gridsearch.py index 24e0a74..7d453c9 100644 --- a/gensvm/gridsearch.py +++ b/gensvm/gridsearch.py @@ -602,6 +602,7 @@ class GenSVMGridSearchCV(BaseEstimator, MetaEstimatorMixin): # gamma='auto' was used. We need to save the actual numerical value # for use in the predict method later on, so we extract that here. if ( + "kernel" in self.best_params_ and not self.best_params_["kernel"] == "linear" and not "gamma" in self.best_params_ ): -- cgit v1.2.3