def test_train_and_topics_categorical(): model = MetaBowEstimator(vocabulary, batch_size=32) model.fit(X_scipy, y_numpy) model.get_topic_vectors() assert (True)
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)
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]))
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)