xapian-letor/letor_internal.h

   1 /** @file letor_internal.h
   2  * @brief Internals of Xapian::Letor class
   3  */
   4 /* Copyright (C) 2011 Parth Gupta
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301
  19  * USA
  20  */
  21
  22 #ifndef XAPIAN_INCLUDED_LETOR_INTERNAL_H
  23 #define XAPIAN_INCLUDED_LETOR_INTERNAL_H
  24
  25 #include <xapian/letor.h>
  26 #include "ranker.h"
  27
  28 #include <map>
  29
  30 using namespace std;
  31
  32 namespace Xapian {
  33
  34 class Letor::Internal : public Xapian::Internal::intrusive_base {
  35     friend class Letor;
  36     Ranker ranker;
  37     Database letor_db;
  38     Query letor_query;
  39
  40   public:
  41
  42     std::map<Xapian::docid, double>  letor_score(const Xapian::MSet & mset);
  43
  44     void letor_learn_model(int svm_type, int kernel_type);
  45
  46     void prepare_training_file(const std::string & query_file, const std::string & qrel_file, Xapian::doccount msetsize);
  47
  48
  49 };
  50
  51 }
  52
  53 #endif // XAPIAN_INCLUDED_LETOR_INTERNAL_H