LinearModel.h

00001 #ifndef __LINEAR_MODEL_
00002 #define __LINEAR_MODEL_
00003 
00004 #include "StandardAlgorithm.h"
00005 #include "Framework.h"
00006 
00016 class LinearModel : public StandardAlgorithm, public Framework
00017 {
00018 public:
00019     LinearModel();
00020     ~LinearModel();
00021 
00022     virtual void modelInit();
00023     virtual void modelUpdate ( REAL* input, REAL* target, uint nSamples, uint crossRun );
00024     virtual void predictAllOutputs ( REAL* rawInputs, REAL* outputs, uint nSamples, uint crossRun );
00025     virtual void readSpecificMaps();
00026     virtual void saveWeights ( int cross );
00027     virtual void loadWeights ( int cross );
00028     virtual void loadMetaWeights ( int cross );
00029 
00030     static string templateGenerator ( int id, string preEffect, int nameID, bool blendStop );
00031 
00032 private:
00033 
00034     // solution weights (per cross validation set)
00035     REAL** m_x;
00036     double m_reg;
00037     bool m_tuneRigeModifiers;
00038 
00039     NumericalTools solver;
00040     double* m_ridgeModifiers;
00041 };
00042 
00043 
00044 #endif