Contextualized Topic Models
Project description
Contextualized Topic Models
Contextualized Topic Models
Free software: MIT license
Documentation: https://contextualized-topic-models.readthedocs.io.
Super big shout-out to Stephen Carrow for creating the awesome https://github.com/estebandito22/PyTorchAVITM package from which we constructed the foundations of this package. We are happy to redistribute again this software under the MIT License.
Features
TODO
Quick Guide
Install the package using pip
pip install -U contextualized_topic_models
The contextual neural topic model can be easily instantiated using few parameters (although there is a wide range of parameters you can use to change the behaviour of the neural topic model.
cotm = COTM(input_size=1000, bert_input_size=512, inferencetype="contextual")
cotm.fit()
See the example notebook in the contextualized_topic_models/examples folder
Team
Federico Bianchi <f.bianchi@unibocconi.it> Bocconi University
Silvia Terragni <s.terragni4@campus.unimib.it> University of Milan-Bicocca
Dirk Hovy <dirk.hovy@unibocconi.it> Bocconi University
Credits
This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template. To ease the use of the library we have also incuded the rbo package, all the rights reserved to the author of that package.
History
0.1.0 (2020-04-04)
First release on PyPI.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for contextualized_topic_models-1.0.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7c56a7797cbaec4b82de0a893b6c793dac344fd9cc637dd2db604f35b06a8c06 |
|
MD5 | d7b5646365926f867b4e1040949b7d3b |
|
BLAKE2b-256 | 511d32a5da861fe0570ea1d3276a9ca70fac0019339e7a8d6d0780660f27b8ae |
Hashes for contextualized_topic_models-1.0.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9777643ad352b34383a7780861bb9d69f7e7ff19fcc0b6c6457f49aa56cb4f7c |
|
MD5 | 5fa68d2f8cee9567f748ce120ddbdf42 |
|
BLAKE2b-256 | 65fdad9f00fb49d53df9aedf63e3c5ff63d1e65bfb98b468f4d29cc5473577c9 |