00001 00004 #ifndef __sgd_model_h 00005 #define __sgd_model_h 00006 00007 #include "vectors.h" 00008 #include <cmath> 00009 #include "globals.h" 00010 #include <algorithm> 00011 #include "model.h" 00012 #include "incremental_sgd.h" 00013 00015 class SvmSgd { 00016 private: 00017 int _dim; 00018 model _current; 00020 int ex; 00022 int match; 00024 Incremental_SGD *sgd; 00025 public: 00031 SvmSgd(int dim, double lambda); 00035 void scale(); 00040 struct model& get_model() { return _current;} 00044 void test_example(char *newfeature); 00049 double getAccuracy(); 00054 bool isExampleCorrectlyClassified(std::string newfeature); 00055 }; 00056 00057 #endif 00058