Method saveClassificationResult

ml/src/svm_wrapper.cpp:842–874

}

void
pcl::SVMClassify::saveClassificationResult(const char* filename)
{
  assert(prediction_.size() > 0);
  assert(model_.l > 0);

  std::ofstream output;
  output.open(filename);

  int nr_class = svm_get_nr_class(&model_);
  int* labels = static_cast<int*>(malloc(nr_class * sizeof(int)));
  svm_get_labels(&model_, labels);

  if (predict_probability_) {
    output << "labels ";

    for (int j = 0; j < nr_class; j++)
      output << labels[j] << " ";

    output << "\n";
  }

  for (const auto& prediction : prediction_) {
    for (const double& value : prediction)
      output << value << " ";

    output << "\n";
  }

  output.close();

  free(labels);
}

#define PCL_INSTANTIATE_SVM(T) template class PCL_EXPORTS pcl::SVM<T>;
#define PCL_INSTANTIATE_SVMTrain(T) template class PCL_EXPORTS pcl::SVMTrain<T>;