CEBL  2.1
SimpleMax.hpp
Go to the documentation of this file.
1 /*
2 * CEBL : CSU EEG Brain-Computer Interface Lab
3 *
4 * Author: Jeshua Bratman - jeshuabratman@gmail.com
5 *
6 * This file is part of CEBL.
7 *
8 * CEBL is free software; you can redistribute it and/or modify it.
9 * We only ask that if you use our code that you cite the source in
10 * your project or publication.
11 *
12 * EEG Group (www.cs.colostate.edu/eeg)
13 * Department of Computer Science
14 * Colorado State University
15 *
16 */
17 
25 #ifndef SimpleMax_H
26 #define SimpleMax_H
27 
28 #include "../model/Decision.hpp"
29 
30 namespace CEBL
31 {
32  class SimpleMax : public Decision
33  {
34  private:
35  std::vector<double> proportions;
36  double growth_rate;
37  double shrink_rate;
38 
39  public:
40  SimpleMax();
41  ~SimpleMax();
42  void init(int num_classes);
43 
44  void updateWithProbabilities(std::vector<double> probs);
45  std::vector<double> decideClasses();
46 
47 
48  //params
49  std::map<std::string, CEBL::Param> getParamsList();
50  void setParamsList( std::map<std::string, CEBL::Param>&);
51 
52  //saving and loading
53  map<string, SerializedObject> save() const;
54  void load(map<string, SerializedObject> objects);
55 
56  };
57 }
58 
59 
60 #endif