Skip to content

Commit b04fce4

Browse files
committed
fix a bug for linear coef
1 parent a7442a3 commit b04fce4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/thundersvm/thundersvm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def fit(self, X, y):
142142
thundersvm.get_rho(rho, rho_size, c_void_p(self.model))
143143

144144
if self.kernel == 'linear':
145-
coef = (c_float * (self.n_binary_model * self.n_sv))()
145+
coef = (c_float * (self.n_binary_model * self.n_features))()
146146
thundersvm.get_linear_coef(coef, self.n_binary_model, self.n_features, c_void_p(self.model))
147147
self.coef_ = np.array([coef[index] for index in range(0, self.n_binary_model * self.n_features)]).astype(float)
148148
self.coef_ = np.reshape(self.coef_, (self.n_binary_model, self.n_features))

0 commit comments

Comments
 (0)