Skip to content

Commit 44fd56b

Browse files
committed
Merge pull request scikit-learn#5093 from amueller/one_class_svm_memory_fix
[MRG+2] Fix memory access error in OneClassSVM
2 parents 250e509 + 7a3a987 commit 44fd56b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn/svm/classes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
LinearModel
88
from ..feature_selection.from_model import _LearntSelectorMixin
99
from ..utils import check_X_y
10+
from ..utils.validation import _num_samples
1011

1112

1213
class LinearSVC(BaseEstimator, LinearClassifierMixin,
@@ -1011,7 +1012,7 @@ def fit(self, X, y=None, sample_weight=None, **params):
10111012
If X is not a C-ordered contiguous array it is copied.
10121013
10131014
"""
1014-
super(OneClassSVM, self).fit(X, [], sample_weight=sample_weight,
1015+
super(OneClassSVM, self).fit(X, np.ones(_num_samples(X)), sample_weight=sample_weight,
10151016
**params)
10161017
return self
10171018

0 commit comments

Comments
 (0)