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_EXPERIMENT_H
#define EXPERIMENT_EXPERIMENT_H

#include <vector>
#include "kernel/matrix.h"
#include "kernel/kernel-matrix.h"
#include "util/labeled-index.h"

using namespace std;

using namespace libpmk;
namespace libpmk_util {


class Experiment {
public:
  Experiment(vector<LabeledIndex> training,
             vector<LabeledIndex> testing,
             const KernelMatrix& kernel);

  Experiment(vector<LabeledIndex> training,
             const KernelMatrix& training_matrix,
             vector<LabeledIndex> testing, const Matrix& testing_matrix);

  virtual ~Experiment() { }

  virtual void Train() = 0;


  virtual int Test() = 0;



  int GetPrediction(int test_index) const;

  int GetNumCorrect() const;

  int GetNumCorrect(int label) const;

  int GetNumTestExamples() const;

  int GetNumTestExamples(int label) const;

  double GetAccuracy() const;

  double GetAccuracy(int label) const;

protected:
  vector<LabeledIndex> training_;
  vector<LabeledIndex> testing_;

  double GetKernelValue(int row, int col) const;

  double GetKernelValue(const LabeledIndex& row,
                        const LabeledIndex& col) const;

  void SetPrediction(int test_index, int prediction);

private:
  KernelMatrix kernel_;
  vector<int> predictions_;
};
}  // namespace libpmk_util
#endif  // EXPERIMENT_EXPERIMENT_H

---