---

CosSimRetMethod.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2002 University of Massachusetts.  All Rights Reserved.
 *
 * Use of the Lemur Toolkit for Language Modeling and Information Retrieval
 * is subject to the terms of the software license set forth in the LICENSE
 * file included with this software, and also available at
 * http://www.cs.cmu.edu/~lemur/license.html
 *
 *==========================================================================
*/
/*
  Author: dmf 9/2002
 */

#ifndef _COSSIMRETMETHOD_HPP
#define _COSSIMRETMETHOD_HPP

#include "TextQueryRetMethod.hpp"

namespace CosSimParameter {
  struct FeedbackParam {
    int howManyTerms;
    double posCoeff;
  };    
  static int defaultHowManyTerms = 50;
  static double defaultPosCoeff = 0.5;
  static const char *defaultL2File = "cos.L2";
};

class CosSimQueryRep : public ArrayQueryRep {
public:
  CosSimQueryRep(TextQuery &qry, Index &dbIndex, double *idfValue);
  CosSimQueryRep(int docId, Index &dbIndex, double *idfValue);
  virtual ~CosSimQueryRep() {}
protected:
  double *idf;
  Index &ind;
};

class CosSimDocRep : public DocumentRep {
public:
  CosSimDocRep(int docID, Index &dbIndex, double *idfValue, double norm) : 
    DocumentRep(docID), ind(dbIndex), idf(idfValue), dNorm(norm) {
  }
  virtual ~CosSimDocRep() { }
  virtual double termWeight(int termID, DocInfo *info) { 
    return (idf[termID]*info->termCount()); 
  }
  virtual double scoreConstant() { return dNorm;}

private:
  Index & ind;
  double *idf;
  double dNorm; 
};

class CosSimScoreFunc : public ScoreFunction {
public:
  CosSimScoreFunc(Index &dbIndex): ind(dbIndex) {}
  virtual double adjustedScore(double origScore, TextQueryRep *qRep, 
                               DocumentRep *dRep) {
    return origScore/dRep->scoreConstant();
  }
protected:
  Index &ind;
};

class CosSimRetMethod : public TextQueryRetMethod {
public:
  CosSimRetMethod(Index &dbIndex, ScoreAccumulator &accumulator);
  CosSimRetMethod(Index &dbIndex, const char *L2file,
                  ScoreAccumulator &accumulator);

  virtual ~CosSimRetMethod();

  virtual TextQueryRep *computeTextQueryRep(TextQuery &qry) {
    return (new CosSimQueryRep(qry, ind, idfV));
  }

  virtual TextQueryRep *computeTextQueryRep(int docid) {
    return (new CosSimQueryRep(docid, ind, idfV));
  }

  virtual DocumentRep *computeDocRep(int docID) { 
    return (new CosSimDocRep(docID, ind, idfV, docNorm(docID)));
  }
  virtual ScoreFunction *scoreFunc() {
    return (scFunc);
  }
  virtual void updateTextQuery(TextQueryRep &qryRep, DocIDSet &relDocs);
  void setFeedbackParam(CosSimParameter::FeedbackParam &feedbackParam);
protected:
  
  double *idfV;
  ScoreFunction *scFunc;
  double * docNorms;
  int numDocs, numTerms;


  CosSimParameter::FeedbackParam fbParam;
  const char *L2FileName;

  double docNorm(int docID);
  void loadDocNorms();
};

inline void CosSimRetMethod::setFeedbackParam(CosSimParameter::FeedbackParam 
                                              &feedbackParam) {
  fbParam = feedbackParam;
}
#endif /* _COSSIMRETMETHOD_HPP */

---