org.ranksys.lda

Class LDAModelEstimator

java.lang.Object

org.ranksys.lda.LDAModelEstimator

public class LDAModelEstimator
extends Object

LDA model estimator. See ParallelTopicModel in Mallet (http://mallet.cs.umass.edu/) for more details.

Author:

Saúl Vargas (Saul.Vargas@glasgow.ac.uk)

Constructor Summary

Constructors

Constructor and Description
LDAModelEstimator()

Method Summary

Modifier and Type	Method and Description
static <U,I> cc.mallet.topics.ParallelTopicModel	estimate(FastPreferenceData<U,I> preferences, int k, double alpha, double beta, int numIterations, int burninPeriod) Estimate a topic model for collaborative filtering data.

Methods inherited from class java.lang.Object

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

Constructor Detail

LDAModelEstimator

public LDAModelEstimator()

Method Detail

estimate

public static <U,I> cc.mallet.topics.ParallelTopicModel estimate(FastPreferenceData<U,I> preferences,
                                                                 int k,
                                                                 double alpha,
                                                                 double beta,
                                                                 int numIterations,
                                                                 int burninPeriod)
                                                          throws IOException

Estimate a topic model for collaborative filtering data.

Type Parameters:

U - user type

I - item type

Parameters:

preferences - preference data

k - number of topics

alpha - alpha in model

beta - beta in model

numIterations - number of iterations

burninPeriod - burnin period

Returns:

a topic model

Throws:

IOException - when internal IO error occurs