Ejemplo n.º 1
0
def test_train_and_topics_categorical():
    model = MetaBowEstimator(vocabulary, batch_size=32)
    model.fit(X_scipy, y_numpy)
    model.get_topic_vectors()
    assert (True)
Ejemplo n.º 2
0
def test_train_and_npmi_categorical():
    model = MetaBowEstimator(vocabulary, batch_size=32)
    model.fit(X_scipy, y_numpy)
    assert (model._npmi_per_covariate(X_scipy, y_numpy, 10) == 0)
Ejemplo n.º 3
0
def test_train_and_transform_categorical():
    model = MetaBowEstimator(vocabulary, batch_size=32)
    model.fit(X_scipy, y_numpy)
    trans = model.transform(X_scipy, y_numpy)
    assert (np.all(trans == trans[0]))
Ejemplo n.º 4
0
def test_train_and_perplexity_categorical():
    model = MetaBowEstimator(vocabulary, batch_size=32)
    model.fit(X_scipy, y_numpy)
    model.perplexity(X_scipy, y_numpy)
    assert (True)