Example #1
0
def test_run():
    run_pipeline(os.path.join(test_data_path, "test_data_json_stream.json"),
                 content_field="abstract",
                 termite_plot=True,
                 lda_vis=False)
    assert os.path.isfile("termite.html")
    os.remove("termite.html")
Example #2
0
File: cli.py Project: dm04806/topik
def run(data, format, output, tokenizer, vectorizer, ntopics, field, model,
        termite, lda_vis):
    run_pipeline(data_source=data,
                 source_type=format,
                 dir_path=output,
                 tokenizer=tokenizer,
                 vectorizer=vectorizer,
                 ntopics=ntopics,
                 content_field=field,
                 model=model,
                 lda_vis=lda_vis,
                 termite_plot=termite)
Example #3
0
def test_run():
    run_pipeline(os.path.join(test_data_path, "test_data_json_stream.json"), content_field="abstract",
                 termite_plot=True, lda_vis=False)
    assert os.path.isfile("termite.html")
    os.remove("termite.html")
Example #4
0
def run(data, format, output, tokenizer, vectorizer, ntopics, field, model,
        termite, lda_vis):
    run_pipeline(data_source=data, source_type=format, dir_path=output,
                 tokenizer=tokenizer, vectorizer=vectorizer, ntopics=ntopics,
                 content_field=field, model=model, lda_vis=lda_vis,
                 termite_plot=termite)