Commit Graph
8 Commits
Author SHA1 Message Date
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
Michael Tu 2f7731b4d5 Update data path (#127) 2018-06-23 23:55:42 -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
Ralph Tang 62f8abec7f Add README (#110)
* Make *QA/MSRVID work with VDPWI

* Add README
2018-05-25 11:52:25 -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
Ralph Tang 494ce36575 Refactor VDPWI to use common API (#109) 2018-05-25 01:30:53 -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