prob-svm-experiment.h

Go to the documentation of this file.

// Copyright 2007, Massachusetts Institute of Technology.
// The use of this code is permitted for research only. There is
// absolutely no warranty for this software.
//
// Author: John Lee (jjl@mit.edu)
//

#ifndef EXPERIMENT_SVM_EXPERIMENT_H
#define EXPERIMENT_SVM_EXPERIMENT_H

#include "experiment/experiment.h"
#include "kernel/kernel-matrix.h"
#include "util/labeled-index.h"
#include "svm/svm.h"

using namespace libpmk;

namespace libpmk_util {


class ProbSVMExperiment : public Experiment {
public:
  ProbSVMExperiment(vector<LabeledIndex> training,
                    vector<LabeledIndex> testing,
                    const KernelMatrix& kernel,
                    double c);
  virtual ~ProbSVMExperiment();

  virtual void Train();
  virtual int Test();

  double GetProbability(int test_index, int which_class);

private:
  struct svm_parameter param_;
  struct svm_model *model_;

  void SetProbability(int test_index, int which_class, double prob);

  // Vector of length <num_test_examples>.
  // Each vector is a vec of length <num_classes>.
  vector<vector<double> > probabilities_;

};
}  // namespace libpmk_util
#endif  // EXPERIMENT_SVM_EXPERIMENT_H

---