Does corex have a predict function?

[RMZ3]

Hello,

I have trained a CorEx model on a set of documents, but I now have new documents I want to infer topics for using the prior model. Is there a way to do this using CorEx?

[gregversteeg]

Yes, there is a predict function. You have to use the python API, though, I don't think there's a way to do it from the command line interface.

[RMZ3]

Thank you. Is there any example code on how to use the predict function?

[GiarteDataTeam]

You can find some samples here

This also may help:

    doc_word = vector.transform(doc)
    words = list(np.asarray(vector.get_feature_names()))
    # final step of preprocessing where we remove all integers from our set of words.
    not_digit_inds = [ind for ind, word in enumerate(words) if not word.isdigit()]
    doc_word = doc_word[:, not_digit_inds]
    topics = corex_model.predict(doc_word)

where vector is defined during training and developing the model

vector = CountVectorizer(stop_words='english',  max_features=max_words, lowercase=True, ngram_range=ngram,
                                 binary=True)