diff options
| -rw-r--r-- | gensvm/gridsearch.py | 1 |
1 files changed, 1 insertions, 0 deletions
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_ ): |