search

MIND-Lab / OCTIS

OCTIS: Comparing Topic Models is Simple! A python package to optimize and evaluate topic models (accepted at EACL2021 demo track)
MIT License
705 stars 98 forks source link

Add vONTSS #111

Open jiangxiaoyuwww opened 9 months ago

jiangxiaoyuwww commented 9 months ago

Tested via dataset loaded via load_dataset 

from datasets import load_dataset
from octis.evaluation_metrics.diversity_metrics import TopicDiversity
from octis.evaluation_metrics.coherence_metrics import Coherence
from octis.models.VONT import VONT

df = load_dataset('xwjzds/ag_news')
model_output = model.train_model(df['train']['text'])# Train the model

TODO: change dataset proprocessing function to take OCTIS's dataset format as input