xapian-letor/featurevector.h

   1 /* featurevector.h: The file responsible for transforming the document into the feature space.
   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 FEATURE_VECTOR_H
  22 #define FEATURE_VECTOR_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 "featuremanager.h"
  31
  32 #include <list>
  33 #include <map>
  34
  35 using namespace std;
  36
  37
  38 namespace Xapian {
  39
  40 class XAPIAN_VISIBILITY_DEFAULT FeatureVector {
  41
  42   public:
  43
  44     double label;
  45     std::map<int,double> fvals;
  46     int fcount;
  47     string did;
  48
  49
  50     FeatureVector();
  51
  52     FeatureVector(const FeatureVector & o);
  53
  54     virtual ~FeatureVector() {};
  55
  56     /** This method takes the document from the MSet as input and gives the feature representation
  57      * as vector in the form of
  58      * map<int,double>
  59      */
  60 //    std::map<int,double> transform(const Xapian::Document & doc);
  61
  62
  63     map<string, map<string, int> > load_relevance(const std::string & qrel_file);
  64
  65 //    void set_qid(const std::string & qid);
  66
  67     void set_did(const std::string & did1);
  68
  69     void set_label(double label1);
  70     void set_fvals(map<int,double> fvals1);
  71
  72 };
  73
  74 }
  75 #endif /* FEATURE_VECTOR_H */