example-selector.h

Go to the documentation of this file.

// Copyright 2006, 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_EXAMPLE_SELECTOR_H
#define EXPERIMENT_EXAMPLE_SELECTOR_H

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

using namespace std;
using namespace libpmk;

namespace libpmk_util {


class ExampleSelector {
 public:
   ExampleSelector(const vector<int>& labels);
   virtual ~ExampleSelector() { }

   vector<LabeledIndex> GetTrainingExamples();

   vector<LabeledIndex> GetTestingExamples();

   vector<int> GetUniqueLabels();

   int GetNumExamples() const;

 protected:
   const vector<LabeledIndex>& GetExamples() const;


   vector<LabeledIndex>
      InvertSelection(const vector<LabeledIndex>& selection) const;


   vector<LabeledIndex>
      InvertSelection(const vector<LabeledIndex>& selection,
                      const vector<LabeledIndex>& superset) const;



   vector<LabeledIndex>
      RandomSample(int n, const vector<LabeledIndex>& selection) const;


   vector<LabeledIndex>
      GetInstancesWithLabel(int label,
                            const vector<LabeledIndex>& selection) const;

   void AddTrainingExample(LabeledIndex index);

   void AddTestingExample(LabeledIndex index);

   virtual void SelectExamples() = 0;

 private:
   // The internal function that manages the state of whether SelectExamples
   // has already been called and sorts the vectors.
   void LoadExamples();
   vector<LabeledIndex> labels_;
   vector<LabeledIndex> training_;
   vector<LabeledIndex> testing_;
   vector<int> unique_labels_;
   bool selected_;
};
}  // namespace libpmk_util
#endif  // EXPERIMENT_EXAMPLE_SELECTOR_H

---