xapian-letor/svmranker.cc

   1 /* ranker.h: The abstract ranker 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/base.h>
  27 #include <xapian/types.h>
  28 #include <xapian/visibility.h>
  29
  30 #include "ranker.h"
  31 #include "ranklist.h"
  32 //#include "evalmetric.h"
  33
  34 #include <list>
  35 #include <map>
  36
  37 using namespace std;
  38
  39
  40 namespace Xapian {
  41
  42 class XAPIAN_VISIBILITY_DEFAULT SVMRanker: public Ranker {
  43
  44     string model = null;
  45     double[] weight;
  46   public:
  47     SVMRanker() {};
  48
  49     /* Override all the four methods below in the ranker sub-classes files
  50      * wiz svmranker.cc , listnet.cc, listmle.cc and so on
  51      */
  52     Xapian::RankList rank(const Xapian::RankList & rl);
  53
  54     void learn_model();
  55
  56     void load_model(const std::string & model_file);
  57
  58     void save_model();
  59
  60     double score(const Xapian::FeatureVector & fv);
  61
  62 };
  63
  64 }
  65 #endif /* SVMRANKER_H */