Skip to content
Snippets Groups Projects
Commit a6ca8195 authored by Guillaume Pasero's avatar Guillaume Pasero
Browse files

ENH: expose method to check if SVM model has probability estimates

parent 18a51239
No related branches found
No related tags found
No related merge requests found
...@@ -307,6 +307,12 @@ public: ...@@ -307,6 +307,12 @@ public:
return static_cast<bool>(m_Parameters.probability); return static_cast<bool>(m_Parameters.probability);
} }
/** Test if the model has probabilities */
bool HasProbabilities(void) const
{
return static_cast<bool>(svm_check_probability_model(m_Model));
}
/** Return number of support vectors */ /** Return number of support vectors */
int GetNumberOfSupportVectors(void) const int GetNumberOfSupportVectors(void) const
{ {
......
...@@ -464,7 +464,7 @@ SVMModel<TValue, TLabel>::EvaluateProbabilities(const MeasurementType& measure) ...@@ -464,7 +464,7 @@ SVMModel<TValue, TLabel>::EvaluateProbabilities(const MeasurementType& measure)
itkExceptionMacro(<< "Model is not up-to-date, can not predict probabilities"); itkExceptionMacro(<< "Model is not up-to-date, can not predict probabilities");
} }
if (svm_check_probability_model(m_Model) == 0) if (!this->HasProbabilities())
{ {
throw itk::ExceptionObject(__FILE__, __LINE__, throw itk::ExceptionObject(__FILE__, __LINE__,
"Model does not support probability estimates", ITK_LOCATION); "Model does not support probability estimates", ITK_LOCATION);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment