dragon.ir.topicmodel

Class TwoDimensionEM

java.lang.Object

dragon.ir.topicmodel.AbstractModel

dragon.ir.topicmodel.AbstractTwoDimensionModel

dragon.ir.topicmodel.TwoDimensionEM

All Implemented Interfaces:

TwoDimensionModel

public class TwoDimensionEM
extends AbstractTwoDimensionModel

The EM based two dimesional topical model

Copyright: Copyright (c) 2005

Company: IST, Drexel University

Version:

1.0

Author:

Davis Zhou

Field Summary

Fields

Modifier and Type	Field and Description
protected double	comThemeCoeffi
protected double	themeBkgCoeffi
protected DoubleVector	themeBkgModel
protected double	viewBkgCoeffi
protected DoubleVector	viewBkgModel

Fields inherited from class dragon.ir.topicmodel.AbstractTwoDimensionModel

arrCommonThemeProb, arrDocTheme, arrDocView, arrThemeProb, arrViewProb, docNum, themeNum, themeTermNum, topicIndexReader, viewIndexReader, viewNum, viewTermNum

Fields inherited from class dragon.ir.topicmodel.AbstractModel

iterations, seed, statusOut

Constructor Summary

Constructors

Constructor and Description
TwoDimensionEM(IndexReader viewIndexReader, DoubleVector viewBkgModel, double viewBkgCoeffi, IndexReader topicIndexReader, DoubleVector themeBkgModel, double themeBkgCoeffi, double comThemeCoeffi)
TwoDimensionEM(IndexReader viewIndexReader, IndexReader topicIndexReader, double viewBkgCoeffi, double themeBkgCoeffi, double comThemeCoeffi)

Method Summary

Modifier and Type	Method and Description
boolean	estimateModel(int viewNum, int topicNum)
protected void	initialize(int docNum, int viewTermNum, int viewNum, double[][] arrViewModel, double[][] arrDocView, int themeTermNum, int themeNum, double[][] arrThemeCommonModel, double[][][] arrThemeModel, double[][][] arrDocTheme)

Methods inherited from class dragon.ir.topicmodel.AbstractTwoDimensionModel

getCommonTopic, getDocNum, getDocTopics, getDocViews, getTopicNum, getTopicTermName, getTopicTermNum, getView, getViewNum, getViewTermName, getViewTermNum, getViewTopic

Methods inherited from class dragon.ir.topicmodel.AbstractModel

getBkgModel, getIterationNum, printStatus, setIterationNum, setRandomSeed, setStatusOut

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

viewBkgModel

protected DoubleVector viewBkgModel

themeBkgModel

protected DoubleVector themeBkgModel

viewBkgCoeffi

protected double viewBkgCoeffi

themeBkgCoeffi

protected double themeBkgCoeffi

comThemeCoeffi

protected double comThemeCoeffi

Constructor Detail

TwoDimensionEM

public TwoDimensionEM(IndexReader viewIndexReader,
                      IndexReader topicIndexReader,
                      double viewBkgCoeffi,
                      double themeBkgCoeffi,
                      double comThemeCoeffi)

TwoDimensionEM

public TwoDimensionEM(IndexReader viewIndexReader,
                      DoubleVector viewBkgModel,
                      double viewBkgCoeffi,
                      IndexReader topicIndexReader,
                      DoubleVector themeBkgModel,
                      double themeBkgCoeffi,
                      double comThemeCoeffi)

Method Detail

estimateModel

public boolean estimateModel(int viewNum,
                             int topicNum)

initialize

protected void initialize(int docNum,
                          int viewTermNum,
                          int viewNum,
                          double[][] arrViewModel,
                          double[][] arrDocView,
                          int themeTermNum,
                          int themeNum,
                          double[][] arrThemeCommonModel,
                          double[][][] arrThemeModel,
                          double[][][] arrDocTheme)