import libsvm_predict_probability and libsvm_predict_values from #76

Co-authored-by: Iblis Lin <iblis@hs.ntnu.edu.tw>
JuliaML · Jun 11, 2021 · c7df763 · c7df763
1 parent e4c64c3
commit c7df763
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/LIBSVM.jl b/src/LIBSVM.jl
@@ -446,6 +446,18 @@ function svmpredict(model::SVM{T}, X::AbstractMatrix{U}; nt::Integer = 0) where
     (pred, decvalues)
 end
 
+function libsvm_predict_probability(model::SVMModel, node::Ptr{SVMNode}, dec::Ref{Cdouble})
+    ccall((:svm_predict_probability, libsvm), Cdouble,
+          (Ref{SVMModel}, Ptr{SVMNode}, Ref{Cdouble}),
+          model, node, dec)
+end
+
+function libsvm_predict_values(model::SVMModel, node::Ptr{SVMNode}, dec::Ref{Cdouble})
+    ccall((:svm_predict_values, libsvm), Cdouble,
+          (Ref{SVMModel}, Ptr{SVMNode}, Ref{Cdouble}),
+          model, node, dec)
+end
+
 include("ScikitLearnTypes.jl")
 include("ScikitLearnAPI.jl")