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

---