diff --git a/sklearn_porter/estimator/classifier/SVC/__init__.py b/sklearn_porter/estimator/classifier/SVC/__init__.py index 3c1ceaac..c71908f1 100644 --- a/sklearn_porter/estimator/classifier/SVC/__init__.py +++ b/sklearn_porter/estimator/classifier/SVC/__init__.py @@ -182,6 +182,8 @@ def export(self, class_name, method_name, export_data=False, self.gamma = self.params['gamma'] if self.gamma == 'auto': self.gamma = 1. / self.n_features + elif self.gamma == 'scale': + self.gamma = est._gamma # computed value of 1 / (n_features * X.var()) self.gamma = self.repr(self.gamma) # Coefficient and degree: