"docs/vscode:/vscode.git/clone" did not exist on "3ee7e11268eb96c44cad3e57d4c8aa870ed2fa02"
Commit edeb4f86 authored by Davis King's avatar Davis King
Browse files

Made the multiclass SVM work with label types other than basic integers and floats.

parent 967b5215
...@@ -77,7 +77,15 @@ namespace dlib ...@@ -77,7 +77,15 @@ namespace dlib
psi.push_back(std::make_pair(dims-1,static_cast<scalar_type>(-1))); psi.push_back(std::make_pair(dims-1,static_cast<scalar_type>(-1)));
// Find which distinct label goes with this psi. // Find which distinct label goes with this psi.
const long label_idx = index_of_max(mat(distinct_labels) == labels[idx]); long label_idx = 0;
for (unsigned long i = 0; i < distinct_labels.size(); ++i)
{
if (distinct_labels[i] == labels[idx])
{
label_idx = i;
break;
}
}
offset_feature_vector(psi, dims*label_idx); offset_feature_vector(psi, dims*label_idx);
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment