Victor Yang
57f53a81b5
Add SSE model ( #168 )
...
* runnable
* add util file
* update readme
* update final layer and add model name
* update argument
* update readme, delete useless args
* fix comments
* fix more comments
2018-12-17 21:04:37 -05:00
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
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
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
Victor Yang
51d8e29525
add NCE to MP-CNN ( #84 )
...
* update nce-sm
* refactor code, update torchtext
* use shared evaluation
* refactor code, use shared data loader
* refactor code
* refactor code
* refactor code according to Michael's great suggestions
* update readme and requirement
* update datasets and readme
* update data loader
* add space between +
* update refactor code
* add nce-mp
* remove duplicate files
* update readme, refactor code according to mp_cnn and delete duplicate code, follow PEP8 standard
* refactor code, add/delete comments
* import exit from sys
2018-01-03 18:12:57 -05:00
rosequ
68e0ef45b2
Util to build w2v pytorch model ( #92 )
...
* util to build w2v pytorch model
* added the code to build the .pt model
2017-12-06 11:34:29 -05:00
Michael Tu
4470983504
TrecQA for MP-CNN ( #77 )
...
* Add TrecQA dataset and modularize MP-CNN infra
* Stylistic improvements
* Fix and warn about trec_eval path issue
* Update README for MP-CNN
* Update incorrect map/mrr
* MP-CNN: address code review comments
* Create common Castor pair Dataset class
* Move map and mrr computation to Castor utils
* Make map mrr utility trec_eval path more general
2017-11-04 18:22:40 -04:00
Victor Yang