example-selector.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_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

---