Victor Yang
5afb845a4b
Add PIT2015 dataset ( #141 )
...
* add prediction/qrel files dump option
* fix comment
* upgrade to torchtext 0.3
* remove ngram
* update device
* minor update
* add pit2015
* revert update torchtext
* revert update torchtext
* revert update torchtext
2018-08-12 18:56:08 -04:00
Michael Tu
8a00f9cdcd
Make Kim CNN ONNX-exportable ( #136 )
...
* Kim CNN - only set embedding for corresponding mode
* Kim CNN ONNX Export
* Specify dummy ONNX input size from command line
2018-08-04 17:30:24 -04:00
Michael Tu
8563ad5976
Kim CNN OOP Refactoring ( #124 )
...
* Nuke obsolete artifacts
* Refactor Kim CNN
* Make kim_cnn a module
* Fix bugs
* Update README
* Add choices to dataset arg
* update for sst2
update sst.py
update sst.py
* Add Kim CNN dataset choices to args.py
* Update tuned SST-1 accuracy
2018-07-03 16:42:31 -04:00
Victor Yang
921a45e7ca
add storing prediction/qrel files option for test set ( #125 )
...
* add prediction/qrel files dump option
* fix comment
2018-06-18 20:43:40 -04:00
Michael Tu
ef05240819
Update MP-CNN Doc with Pre-Trained Models ( #111 )
...
* Delete outdated troubleshooting section
* QAEvalutor bugfix
* Add instructions for pre-trained models
2018-05-25 11:51:19 -04:00
Michael Tu
d7a631b0a9
MP-CNN with Bugs Fixed and PyTorch v0.4 ( #107 )
...
* Refactor datasets
* Update evaluators
* Update trainers
* Update main and MP-CNN model
* Add serialization util
* Fix bugs
* Refactoring for NCE to use new parent class
2018-05-24 23:42:13 -04:00
Ralph Tang
fbd8629ca2
Move MPCNN API to common module ( #105 )
2018-05-24 18:02:37 -04:00
Victor Yang