New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Add vONTSS #111

Open

jiangxiaoyuwww wants to merge 2 commits into MIND-Lab:master from jiangxiaoyuwww:vontss

jiangxiaoyuwww commented Sep 28, 2023 •

edited

Loading

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

jiangxiaoyuwww added 2 commits

September 28, 2023 22:16

4e68dbd


          fix kl divegence not found error

4c0e2ca

# for free to join this conversation on GitHub. Already have an account? # to comment

Labels

None yet