DETM

DETM copied to clipboard

Custom Dataset

4

Hi, Thank you for sharing your experimentation. I wanted to try your experiment on a custom dataset, and I was wondering how to do it. Should I just modify the...

Shiro-LK

Traceback (most recent call last): File "main.py", line 480, in train(epoch) File "main.py", line 217, in train loss, nll, kl_alpha, kl_eta, kl_theta = model(data_batch, normalized_data_batch, times_batch, train_rnn_inp, args.num_docs_train) File "/home/rodrigo/anaconda3/envs/DETM/lib/python3.6/site-packages/torch/nn/modules/module.py",...

rodrigozzh

Hi, is there a way to infer topic distributions on unseen documents (not used to train the model)? Thanks in advance!

Pierre-datascience