如何在Gensim中使用槌设置主题模型的随机种子?

时间:2018-10-11 14:42:14

标签: python gensim topic-modeling mallet

我一直试图通过使用槌作为gensim中的库来保持主题建模输出的稳定性。但是,我发现木槌可以设置随机种子,但是在gensim中看不到任何参数可以设置它。

2 个答案:

答案 0 :(得分:2)

这刚刚被添加到ldamallet.py包装器中

    def __init__(self, mallet_path, corpus=None, num_topics=100, alpha=50, id2word=None, workers=4, prefix=None,
             optimize_interval=0, iterations=1000, topic_threshold=0.0, random_seed=0):
    """

    Parameters
    ----------
    mallet_path : str
        Path to the mallet binary, e.g. `/home/username/mallet-2.0.7/bin/mallet`.
    corpus : iterable of iterable of (int, int), optional
        Collection of texts in BoW format.
    num_topics : int, optional
        Number of topics.
    alpha : int, optional
        Alpha parameter of LDA.
    id2word : :class:`~gensim.corpora.dictionary.Dictionary`, optional
        Mapping between tokens ids and words from corpus, if not specified - will be inferred from `corpus`.
    workers : int, optional
        Number of threads that will be used for training.
    prefix : str, optional
        Prefix for produced temporary files.
    optimize_interval : int, optional
        Optimize hyperparameters every `optimize_interval` iterations
        (sometimes leads to Java exception 0 to switch off hyperparameter optimization).
    iterations : int, optional
        Number of training iterations.
    topic_threshold : float, optional
        Threshold of the probability above which we consider a topic.
    random_seed: int, optional
        Random seed to ensure consistent results, if 0 - use system clock.

    """

答案 1 :(得分:0)

我遇到了同样的问题,但是要使用最新版本的gensim,这有点棘手。正如 Chris 所说,新版本已实现,但运行起来对我来说很麻烦。请确保执行以下操作,因为您可能正在使用旧的包装器:

  1. conda install -c conda-forge gensim
  2. pip install --upgrade gensim

第二步完成工作,仅安装它不会更新gensim,因为我对此有疑问。

以下链接针对您的问题提供了更多信息:

Gensim Installation

Mallet Wrapper