dragon.ir.topicmodel

Class CrossMixtureModel

java.lang.Object

dragon.ir.topicmodel.AbstractModel

dragon.ir.topicmodel.CrossMixtureModel

public class CrossMixtureModel
extends AbstractModel

Cross mixture topic 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 IntSparseMatrix[]	arrTopicReader
protected double	bkgCoefficient
protected double[]	bkgModel
protected int	collectionNum
protected double	comCoefficient
protected int	maxDocNum
protected int	maxTermNum
protected int	themeNum

Fields inherited from class dragon.ir.topicmodel.AbstractModel

iterations, seed, statusOut

Constructor Summary

Constructors

Constructor and Description
CrossMixtureModel(IntSparseMatrix[] arrTopicMatrix, int themeNum, double[] bkgModel, double bkgCoefficient, double comCoefficient)

Method Summary

Modifier and Type	Method and Description
boolean	estimateModel()
double[][]	getCommonModels()
double[][][]	getDocMemberships()
double[][][]	getModels()
protected void	initialize(int maxTermNum, int collectionNum, int themeNum, int maxDocNum, double[][] arrCommonModel, double[][][] arrModel, double[][][] arrDocMembership)

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

arrTopicReader

protected IntSparseMatrix[] arrTopicReader

bkgModel

protected double[] bkgModel

bkgCoefficient

protected double bkgCoefficient

comCoefficient

protected double comCoefficient

themeNum

protected int themeNum

collectionNum

protected int collectionNum

maxTermNum

protected int maxTermNum

maxDocNum

protected int maxDocNum

Constructor Detail

CrossMixtureModel

public CrossMixtureModel(IntSparseMatrix[] arrTopicMatrix,
                         int themeNum,
                         double[] bkgModel,
                         double bkgCoefficient,
                         double comCoefficient)

Method Detail

getModels

public double[][][] getModels()

getCommonModels

public double[][] getCommonModels()

getDocMemberships

public double[][][] getDocMemberships()

estimateModel

public boolean estimateModel()

initialize

protected void initialize(int maxTermNum,
                          int collectionNum,
                          int themeNum,
                          int maxDocNum,
                          double[][] arrCommonModel,
                          double[][][] arrModel,
                          double[][][] arrDocMembership)