aboutsummaryrefslogtreecommitdiff
path: root/gensvm
diff options
context:
space:
mode:
authorGertjan van den Burg <gertjanvandenburg@gmail.com>2019-03-06 23:04:50 -0500
committerGertjan van den Burg <gertjanvandenburg@gmail.com>2019-03-06 23:04:50 -0500
commitb111c2640a3e270a13af6967bb0f35c94c30eb5c (patch)
treea09d3af7f41466368ac67e4546ed1b1403d7bed2 /gensvm
parentThrow a warning on detecting lapack dependency (diff)
downloadpygensvm-b111c2640a3e270a13af6967bb0f35c94c30eb5c.tar.gz
pygensvm-b111c2640a3e270a13af6967bb0f35c94c30eb5c.zip
bugfix for gamma in kernel
Diffstat (limited to 'gensvm')
-rw-r--r--gensvm/gridsearch.py1
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_
):