public class DistributedLDAModel extends LDAModel
Distributed model fitted by LDA
.
This type of model is currently only produced by Expectation-Maximization (EM).
This model stores the inferred topics, the full training dataset, and the topic distribution for each training document.
param: oldLocalModelOption Used to implement oldLocalModel
as a lazy val, but keeping
copy()
cheap.
Modifier and Type | Method and Description |
---|---|
static IntParam |
checkpointInterval() |
IntParam |
checkpointInterval()
Param for set checkpoint interval (>= 1) or disable checkpoint (-1).
|
static Params |
clear(Param<?> param) |
DistributedLDAModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
void |
deleteCheckpointFiles()
:: DeveloperApi ::
|
static Dataset<Row> |
describeTopics() |
static Dataset<Row> |
describeTopics(int maxTermsPerTopic) |
static DoubleArrayParam |
docConcentration() |
DoubleArrayParam |
docConcentration()
Concentration parameter (commonly named "alpha") for the prior placed on documents'
distributions over topics ("theta").
|
static Vector |
estimatedDocConcentration() |
static String |
explainParam(Param<?> param) |
static String |
explainParams() |
static ParamMap |
extractParamMap() |
static ParamMap |
extractParamMap(ParamMap extra) |
static Param<String> |
featuresCol() |
Param<String> |
featuresCol()
Param for features column name.
|
static <T> scala.Option<T> |
get(Param<T> param) |
String[] |
getCheckpointFiles()
:: DeveloperApi ::
|
static int |
getCheckpointInterval() |
int |
getCheckpointInterval() |
static <T> scala.Option<T> |
getDefault(Param<T> param) |
static double[] |
getDocConcentration() |
double[] |
getDocConcentration() |
static String |
getFeaturesCol() |
String |
getFeaturesCol() |
static int |
getK() |
int |
getK() |
static boolean |
getKeepLastCheckpoint() |
boolean |
getKeepLastCheckpoint() |
static double |
getLearningDecay() |
double |
getLearningDecay() |
static double |
getLearningOffset() |
double |
getLearningOffset() |
static int |
getMaxIter() |
int |
getMaxIter() |
Vector |
getOldDocConcentration()
Get docConcentration used by spark.mllib LDA
|
LDAOptimizer |
getOldOptimizer() |
double |
getOldTopicConcentration()
Get topicConcentration used by spark.mllib LDA
|
static boolean |
getOptimizeDocConcentration() |
boolean |
getOptimizeDocConcentration() |
static String |
getOptimizer() |
String |
getOptimizer() |
static <T> T |
getOrDefault(Param<T> param) |
static Param<Object> |
getParam(String paramName) |
static long |
getSeed() |
long |
getSeed() |
static double |
getSubsamplingRate() |
double |
getSubsamplingRate() |
static double |
getTopicConcentration() |
double |
getTopicConcentration() |
static String |
getTopicDistributionCol() |
String |
getTopicDistributionCol() |
static <T> boolean |
hasDefault(Param<T> param) |
static boolean |
hasParam(String paramName) |
static boolean |
hasParent() |
static boolean |
isDefined(Param<?> param) |
boolean |
isDistributed()
Indicates whether this instance is of type
DistributedLDAModel |
static boolean |
isSet(Param<?> param) |
static IntParam |
k() |
IntParam |
k()
Param for the number of topics (clusters) to infer.
|
static BooleanParam |
keepLastCheckpoint() |
BooleanParam |
keepLastCheckpoint()
For EM optimizer only:
optimizer = "em". |
static DoubleParam |
learningDecay() |
DoubleParam |
learningDecay()
For Online optimizer only:
optimizer = "online". |
static DoubleParam |
learningOffset() |
DoubleParam |
learningOffset()
For Online optimizer only:
optimizer = "online". |
static DistributedLDAModel |
load(String path) |
static double |
logLikelihood(Dataset<?> dataset) |
static double |
logPerplexity(Dataset<?> dataset) |
double |
logPrior()
Log probability of the current parameter estimate:
log P(topics, topic distributions for docs | Dirichlet hyperparameters)
|
static IntParam |
maxIter() |
IntParam |
maxIter()
Param for maximum number of iterations (>= 0).
|
static BooleanParam |
optimizeDocConcentration() |
BooleanParam |
optimizeDocConcentration()
For Online optimizer only (currently):
optimizer = "online". |
static Param<String> |
optimizer() |
Param<String> |
optimizer()
Optimizer or inference algorithm used to estimate the LDA model.
|
static Param<?>[] |
params() |
static void |
parent_$eq(Estimator<M> x$1) |
static Estimator<M> |
parent() |
static MLReader<DistributedLDAModel> |
read() |
static void |
save(String path) |
static LongParam |
seed() |
LongParam |
seed()
Param for random seed.
|
static <T> Params |
set(Param<T> param,
T value) |
static LDAModel |
setFeaturesCol(String value) |
static M |
setParent(Estimator<M> parent) |
static LDAModel |
setSeed(long value) |
static DoubleParam |
subsamplingRate() |
DoubleParam |
subsamplingRate()
For Online optimizer only:
optimizer = "online". |
static String[] |
supportedOptimizers() |
String[] |
supportedOptimizers()
Supported values for Param
optimizer . |
LocalLDAModel |
toLocal()
Convert this distributed model to a local representation.
|
static DoubleParam |
topicConcentration() |
DoubleParam |
topicConcentration()
Concentration parameter (commonly named "beta" or "eta") for the prior placed on topics'
distributions over terms.
|
static Param<String> |
topicDistributionCol() |
Param<String> |
topicDistributionCol()
Output column with estimates of the topic mixture distribution for each document (often called
"theta" in the literature).
|
static Matrix |
topicsMatrix() |
static String |
toString() |
double |
trainingLogLikelihood()
Log likelihood of the observed tokens in the training set,
given the current parameter estimates:
log P(docs | topics, topic distributions for docs, Dirichlet hyperparameters)
|
static Dataset<Row> |
transform(Dataset<?> dataset) |
static Dataset<Row> |
transform(Dataset<?> dataset,
ParamMap paramMap) |
static Dataset<Row> |
transform(Dataset<?> dataset,
ParamPair<?> firstParamPair,
ParamPair<?>... otherParamPairs) |
static Dataset<Row> |
transform(Dataset<?> dataset,
ParamPair<?> firstParamPair,
scala.collection.Seq<ParamPair<?>> otherParamPairs) |
static StructType |
transformSchema(StructType schema) |
static String |
uid() |
StructType |
validateAndTransformSchema(StructType schema)
Validates and transforms the input schema.
|
static void |
validateParams() |
static int |
vocabSize() |
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
describeTopics, describeTopics, estimatedDocConcentration, logLikelihood, logPerplexity, setFeaturesCol, setSeed, topicsMatrix, transform, transformSchema, uid, vocabSize
transform, transform, transform
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn, validateParams
toString
save
public static MLReader<DistributedLDAModel> read()
public static DistributedLDAModel load(String path)
public static String toString()
public static Param<?>[] params()
public static void validateParams()
public static String explainParam(Param<?> param)
public static String explainParams()
public static final boolean isSet(Param<?> param)
public static final boolean isDefined(Param<?> param)
public static boolean hasParam(String paramName)
public static Param<Object> getParam(String paramName)
public static final <T> scala.Option<T> get(Param<T> param)
public static final <T> T getOrDefault(Param<T> param)
public static final <T> scala.Option<T> getDefault(Param<T> param)
public static final <T> boolean hasDefault(Param<T> param)
public static final ParamMap extractParamMap()
public static Dataset<Row> transform(Dataset<?> dataset, ParamPair<?> firstParamPair, scala.collection.Seq<ParamPair<?>> otherParamPairs)
public static Dataset<Row> transform(Dataset<?> dataset, ParamPair<?> firstParamPair, ParamPair<?>... otherParamPairs)
public static Estimator<M> parent()
public static void parent_$eq(Estimator<M> x$1)
public static M setParent(Estimator<M> parent)
public static boolean hasParent()
public static final Param<String> featuresCol()
public static final String getFeaturesCol()
public static final IntParam maxIter()
public static final int getMaxIter()
public static final LongParam seed()
public static final long getSeed()
public static final IntParam checkpointInterval()
public static final int getCheckpointInterval()
public static final IntParam k()
public static int getK()
public static final DoubleArrayParam docConcentration()
public static double[] getDocConcentration()
public static final DoubleParam topicConcentration()
public static double getTopicConcentration()
public static final String[] supportedOptimizers()
public static final Param<String> optimizer()
public static String getOptimizer()
public static final Param<String> topicDistributionCol()
public static String getTopicDistributionCol()
public static final DoubleParam learningOffset()
public static double getLearningOffset()
public static final DoubleParam learningDecay()
public static double getLearningDecay()
public static final DoubleParam subsamplingRate()
public static double getSubsamplingRate()
public static final BooleanParam optimizeDocConcentration()
public static boolean getOptimizeDocConcentration()
public static final BooleanParam keepLastCheckpoint()
public static boolean getKeepLastCheckpoint()
public static void save(String path) throws java.io.IOException
java.io.IOException
public static String uid()
public static int vocabSize()
public static LDAModel setFeaturesCol(String value)
public static LDAModel setSeed(long value)
public static StructType transformSchema(StructType schema)
public static Vector estimatedDocConcentration()
public static Matrix topicsMatrix()
public static double logLikelihood(Dataset<?> dataset)
public static double logPerplexity(Dataset<?> dataset)
public LocalLDAModel toLocal()
WARNING: This involves collecting a large topicsMatrix
to the driver.
public DistributedLDAModel copy(ParamMap extra)
Params
defaultCopy()
.public boolean isDistributed()
LDAModel
DistributedLDAModel
isDistributed
in class LDAModel
public double trainingLogLikelihood()
Notes:
- This excludes the prior; for that, use logPrior
.
- Even with logPrior
, this is NOT the same as the data log likelihood given the
hyperparameters.
- This is computed from the topic distributions computed during training. If you call
logLikelihood()
on the same training dataset, the topic distributions will be computed
again, possibly giving different results.
public double logPrior()
public String[] getCheckpointFiles()
If using checkpointing and LDA.keepLastCheckpoint
is set to true, then there may be
saved checkpoint files. This method is provided so that users can manage those files.
Note that removing the checkpoints can cause failures if a partition is lost and is needed
by certain DistributedLDAModel
methods. Reference counting will clean up the checkpoints
when this model and derivative data go out of scope.
public void deleteCheckpointFiles()
Remove any remaining checkpoint files from training.
getCheckpointFiles
public MLWriter write()
MLWritable
MLWriter
instance for this ML instance.public IntParam k()
public int getK()
public DoubleArrayParam docConcentration()
This is the parameter to a Dirichlet distribution, where larger values mean more smoothing (more regularization).
If not set by the user, then docConcentration is set automatically. If set to
singleton vector [alpha], then alpha is replicated to a vector of length k in fitting.
Otherwise, the docConcentration
vector must be length k.
(default = automatic)
Optimizer-specific parameter settings:
- EM
- Currently only supports symmetric distributions, so all values in the vector should be
the same.
- Values should be > 1.0
- default = uniformly (50 / k) + 1, where 50/k is common in LDA libraries and +1 follows
from Asuncion et al. (2009), who recommend a +1 adjustment for EM.
- Online
- Values should be >= 0
- default = uniformly (1.0 / k), following the implementation from
https://github.com/Blei-Lab/onlineldavb
.
public double[] getDocConcentration()
public Vector getOldDocConcentration()
public DoubleParam topicConcentration()
This is the parameter to a symmetric Dirichlet distribution.
Note: The topics' distributions over terms are called "beta" in the original LDA paper by Blei et al., but are called "phi" in many later papers such as Asuncion et al., 2009.
If not set by the user, then topicConcentration is set automatically. (default = automatic)
Optimizer-specific parameter settings:
- EM
- Value should be > 1.0
- default = 0.1 + 1, where 0.1 gives a small amount of smoothing and +1 follows
Asuncion et al. (2009), who recommend a +1 adjustment for EM.
- Online
- Value should be >= 0
- default = (1.0 / k), following the implementation from
https://github.com/Blei-Lab/onlineldavb
.
public double getTopicConcentration()
public double getOldTopicConcentration()
public String[] supportedOptimizers()
optimizer
.public Param<String> optimizer()
For details, see the following papers:
- Online LDA:
Hoffman, Blei and Bach. "Online Learning for Latent Dirichlet Allocation."
Neural Information Processing Systems, 2010.
http://www.cs.columbia.edu/~blei/papers/HoffmanBleiBach2010b.pdf
- EM:
Asuncion et al. "On Smoothing and Inference for Topic Models."
Uncertainty in Artificial Intelligence, 2009.
http://arxiv.org/pdf/1205.2662.pdf
public String getOptimizer()
public Param<String> topicDistributionCol()
This uses a variational approximation following Hoffman et al. (2010), where the approximate distribution is called "gamma." Technically, this method returns this approximation "gamma" for each document.
public String getTopicDistributionCol()
public DoubleParam learningOffset()
optimizer
= "online".
A (positive) learning parameter that downweights early iterations. Larger values make early iterations count less. This is called "tau0" in the Online LDA paper (Hoffman et al., 2010) Default: 1024, following Hoffman et al.
public double getLearningOffset()
public DoubleParam learningDecay()
optimizer
= "online".
Learning rate, set as an exponential decay rate. This should be between (0.5, 1.0] to guarantee asymptotic convergence. This is called "kappa" in the Online LDA paper (Hoffman et al., 2010). Default: 0.51, based on Hoffman et al.
public double getLearningDecay()
public DoubleParam subsamplingRate()
optimizer
= "online".
Fraction of the corpus to be sampled and used in each iteration of mini-batch gradient descent, in range (0, 1].
Note that this should be adjusted in synch with LDA.maxIter
so the entire corpus is used. Specifically, set both so that
maxIterations * miniBatchFraction >= 1.
Note: This is the same as the miniBatchFraction
parameter in
OnlineLDAOptimizer
.
Default: 0.05, i.e., 5% of total documents.
public double getSubsamplingRate()
public BooleanParam optimizeDocConcentration()
optimizer
= "online".
Indicates whether the docConcentration (Dirichlet parameter for document-topic distribution) will be optimized during training. Setting this to true will make the model more expressive and fit the training data better. Default: false
public boolean getOptimizeDocConcentration()
public BooleanParam keepLastCheckpoint()
optimizer
= "em".
If using checkpointing, this indicates whether to keep the last checkpoint. If false, then the checkpoint will be deleted. Deleting the checkpoint can cause failures if a data partition is lost, so set this bit with care. Note that checkpoints will be cleaned up via reference counting, regardless.
See DistributedLDAModel.getCheckpointFiles
for getting remaining checkpoints and
DistributedLDAModel.deleteCheckpointFiles
for removing remaining checkpoints.
Default: true
public boolean getKeepLastCheckpoint()
public StructType validateAndTransformSchema(StructType schema)
schema
- input schemapublic LDAOptimizer getOldOptimizer()
public Param<String> featuresCol()
public String getFeaturesCol()
public IntParam maxIter()
public int getMaxIter()
public LongParam seed()
public long getSeed()
public IntParam checkpointInterval()
public int getCheckpointInterval()