Skip to content

Commit 57a4f21

Browse files
committed
sklearn SVM: actually pass gamma to SVC
1 parent 8592b6f commit 57a4f21

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/svm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def get_optimal_cache_size(n_features, dtype=np.double, max_cache=64):
7373
# Create our C-SVM classifier
7474
clf = SVC(C=params.C, kernel=params.kernel, max_iter=params.maxiter,
7575
cache_size=params.cache_size_mb, tol=params.tol,
76-
shrinking=params.shrinking, gamma='auto')
76+
shrinking=params.shrinking, gamma=params.gamma)
7777

7878
columns = ('batch', 'arch', 'prefix', 'function', 'threads', 'dtype', 'size',
7979
'kernel', 'cache_size_mb', 'C', 'sv_len', 'n_classes', 'accuracy',

0 commit comments

Comments
 (0)