reservoir.h

Go to the documentation of this file.

#ifndef _reservoir_h_
#define _reservoir_h_
#include "vectors.h"
#include "globals.h"

template <class T>
class Reservoir {
 private:
  int reservoir_size;
  int current_size;
  int *exampleClasses;
  T *pFeatures;
  void swapClassesOfLabels(int i, int j);
  void swapVectors(int i, int j);
  void copyExample(int index, int classOfExample, T featureArray);
  void shuffle();
  
  void init(int res_size, long seed);   
 public:

  Reservoir(int res_size);
  Reservoir(int res_size, long seed);
  void addExample(std::string newEx);
  
  void addExample(int y, T ex);
  int getExampleClass(int index);
  const T getExampleFeatureVector(int index);
  int getCurrentSize();
  bool isFull() { return getCurrentSize() ==  reservoir_size; }
};
#include "reservoir.hxx"
#endif

---