xapian-letor/ranker/svmranker.h

   1 /* svmranker.h: The svmranker file.
   2  *
   3  * Copyright (C) 2012 Parth Gupta
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License as
   7  * published by the Free Software Foundation; either version 2 of the
   8  * License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301
  18  * USA
  19  */
  20
  21 #ifndef SVMRANKER_H
  22 #define SVMRANKER_H
  23
  24
  25 #include <xapian.h>
  26 #include <xapian/intrusive_ptr.h>
  27 #include <xapian/types.h>
  28 #include <xapian/visibility.h>
  29
  30 #include "xapian-letor/ranker.h"
  31 #include "xapian-letor/ranklist.h"
  32 #include "xapian-letor/featurevector.h"
  33 //#include "evalmetric.h"
  34
  35 #include <list>
  36 #include <map>
  37
  38 using namespace std;
  39
  40
  41 namespace Xapian {
  42
  43 class XAPIAN_VISIBILITY_DEFAULT SVMRanker: public Ranker {
  44
  45     string model;
  46     double weight[];
  47   public:
  48     SVMRanker() {};
  49
  50     /* Override all the four methods below in the ranker sub-classes files
  51      * wiz svmranker.cc , listnet.cc, listmle.cc and so on
  52      */
  53     std::list<double> rank(const Xapian::RankList & rl);
  54
  55     void learn_model();
  56
  57     void load_model(const std::string & model_file);
  58
  59     void save_model();
  60
  61     double score(const Xapian::FeatureVector & fv);
  62
  63 };
  64
  65 }
  66 #endif /* SVMRANKER_H */