Skip to content

Commit d8e494a

Browse files
committed
fix two minor issues
1 parent acb44c5 commit d8e494a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/thundersvm/thundersvm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,6 @@ def _sparse_decision_function(self, X):
403403
thundersvm.sparse_decision(
404404
X.shape[0], data, indptr, indices,
405405
c_void_p(self.model), dec_size, dec_value_ptr)
406-
self.dec_values = np.array([dec_value_ptr[index] for index in range(0, dec_size)])
407406
self.dec_values = np.frombuffer(dec_value_ptr, dtype=np.float32)\
408407
.reshape((X.shape[0], self.n_binary_model))
409408
return self.dec_values
@@ -430,7 +429,7 @@ def load_from_file(self, path):
430429
self.n_classes = n_classes[0]
431430
n_support_ = (c_int * self.n_classes)()
432431
thundersvm.get_support_classes(n_support_, self.n_classes, c_void_p(self.model))
433-
self.n_support_ = np.frombuffer(n_support_).astype(int)
432+
self.n_support_ = np.frombuffer(n_support_, dtype=np.int32).astype(int)
434433
self.n_sv = thundersvm.n_sv(c_void_p(self.model))
435434

436435
n_feature = (c_int * 1)()

0 commit comments

Comments
 (0)