Cooperative Neural Networks - CoNN
Cooperative Neural Networks(CoNN) : Exploiting prior independence structure for improved classification
(also known as Joint Constraint Networks).
This work is published in NIPS 2018 link.
I am working on a Blog to highlight the novelties and main contributions of this work.
Dependencies
The file 'main.py' is tested on the following
- Python 3
- Pytorch 0.2.0
- Numpy 1.15.1
- nltk (preprocessing text)
- P100 GPUs
Running CoNN-sLDA for Multi-Domain Sentiment Dataset
Getting the data link:
$ wget http://www.cs.jhu.edu/~mdredze/datasets/sentiment/unprocessed.tar.gz
$ tar -xvzf unprocessed.tar.gz
Running the script:
$ python main.py
with the default argparse settings, I get roughly
Avg auc(5 fold CV) = 0.9213456403902894 with std dev = 0.007758676219792392
Development
Code
I will be updating the repo with the following additions
- Script compatible with latest Pytorch version
- Script for latest Tensorflow version
Contributing
Issues can be reported at issues section.
If you want to discuss or contribute, please feel free to drop a mail or raise an issue :)
Collaboration
I will be happy to discuss and collaborate, if you want to use CoNN or its variant for some other Graphical models!
License
CoNN is released under Apache License. You can read about our license at here