Training model on own dataset

adjidieng / ETM

Topic Modeling in Embedding Spaces

MIT License

538 stars 126 forks source link

Training model on own dataset #6

Closed srossgupta closed 4 years ago

srossgupta commented 4 years ago

Is there a way I can train the model on my own dataset ? I am hitting nan on my performance metrics recon_loss, kld_theta = model(data_batch, normalized_data_batch) in main.py

adjidieng commented 4 years ago

Hi there,

We added the script to pre-process your own dataset. Make sure you follow that script. Let us know if you have other questions.