src/PlsaMap.h

/*  -*- C++ -*-

    textIR - A fast text document retrieval engine

    Copyright (C) 2005-2009 Laurence Park

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

    File information:
    $Header: /home/staff/lapark/cvsroot/web_search/textIR/src/PlsaMap.h,v 1.7.2.2 2009/11/05 01:46:39 lapark Exp $
*/

#ifndef PLSAMAP_H
#define PLSAMAP_H

#include "WordSubIndex.h"
#include <cstdlib>
#include <cmath>
#include <ctime>
#include <iostream>

class PlsaMap {
public:

  PlsaMap(int documents, int words, int topics);
  virtual ~PlsaMap(void);

  void printEigenValues(void);

  void deleteTempBuffers(void);

  void initialise(WordSubIndex *index);

  void zeroTempBuffers(void);

  void randomiseRealBuffers(WordSubIndex *index);

  void initialiseRealBuffers(WordSubIndex *index);

  void iterate(void);

  void iterate(int **index);

  void finalise(void);

  void printFile(void);

  void checkProbabilities(void);

  void operator+=(PlsaMap *x);

  virtual void nextElement(void) = 0;

  virtual void reset(void) = 0;

  virtual int term(void) = 0;

  virtual int document(void) = 0;

  virtual float frequency(void) = 0;

  float **Pwz;

  float **Pdz;

  float *Pz; 

  float _R;

  const int topics;
private:
  const int documents;
  const int words;


  bool tempBufferState;

  int iterations;
  float Pzdwt;
  float Pzdwtemp, zDiff, lastzDiff;
  float **nPwz;
  float **nPdz;
  float *nPz, *Pzdw;
};

#endif

---