HMLP: High-performance Machine Learning Primitives
model.hpp
1 #ifndef MODEL_HPP
2 #define MODEL_HPP
3 
4 namespace hmlp
5 {
6 namespace model
7 {
8 
9 template<typename FUNC, typename T>
10 class Classification : public VirtualModel<T>
11 {
12  public:
13 
15  {
16  };
17 
18  void Fit()
19  {
20  };
21 
22  std::vector<T> Predict( std::vector<T> & x )
23  {
24  std::vector<T> predict;
25 
26  return predict;
27  };
28 
29  private:
30 };
31 
32 
33 template<typename FUNC, typename PARAM, typename DATA, typename T>
34 class Regression : public VirtualModel<T>
35 {
36  public:
37 
38  Regression( FUNC *objective, PARAM *param, DATA *data )
39  {
40  };
41 
42  void Fit()
43  {
44  };
45 
46  std::vector<T> Predict( std::vector<T> & x )
47  {
48  std::vector<T> predict;
49 
50  return predict;
51  };
52 
53  private:
54 
58  FUNC *objective;
59 
61  PARAM *param;
62 
63  DATA *data;
64 
65 };
68 template<typename T>
70 {
71  public:
72 
73  VirtualModel() {};
74 
75  virtual void Fit() = 0;
76 
77  virtual std::vector<T> Predict( std::vector<T> & x ) = 0;
78 
79  private:
80 
81 };
82 
83 };
84 };
87 #endif
Definition: model.hpp:69
Definition: model.hpp:34
Definition: model.hpp:10
Definition: gofmm.hpp:83