copilot/OpenSeeFace
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added support for probability inference in ThunderSVMWrapper.

Browse source
This commit is contained in:
Emiliana 2020-09-01 18:50:56 +02:00
parent e854036e56
commit 54cc31a1ef
2 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Source/ThunderSVMWrapper.cpp
View file

@ -149,6 +149,12 @@ extern "C" {
instances.clear();
for (int r = 0; r < rows; r++)
predictions[r] = (float)predicted[r];
if (probabilities != NULL && sw->param.probability) {
vector<float> probs = sw->model->get_prob_predict();
for (int r = 0; r < rows; r++)
for (int c = 0; c < sw->classes; c++)
probabilities[c + r * sw->classes] = probs[c + r * sw->classes];
}
predicted.clear();
return;