wassname/Castor
1
0 Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-10-05 12:20:28 +08:00
Code Issues Packages Projects Releases Wiki Activity
124 Commits 6 Branches 0 Tags
ed4f01852e8977f74d9d2a63fc06097ce29d2183
Commit Graph
15 Commits
Author SHA1 Message Date
Linqing Liu 7dec34a106 Add sts2014 and quora evaluators in common/evaluators/ (#151) 
* add SNLI dataset

* add STS-2014

* add STS-2014

* add trainers and evaluators for Quora

* add quora in datasets/

* process the  merge confict in common/dataset.py

* add sts2014_evaluator.py and quora_evaluator.py
 2018-10-09 09:10:37 -04:00
Linqing Liu de56206fe4 add SNLI / STS-2014 / Quora dataset (#148) 
* add SNLI dataset

* add STS-2014

* add trainers and evaluators for Quora

* add quora in datasets/

* process the  merge confict in common/dataset.py
 2018-10-08 19:23:22 -04:00
Ashutosh-Adhikari 999e0c88f2 Add Reuters dataset option for common.dataset (#149) 
* Add Reuters option in common.dataset
 2018-10-07 00:57:38 -04:00
Achyudh Ram 1b817d3e24 WIP: Add Reuters-21578 dataset (#147) 
* Add ReutersTrainer, ReutersEvaluator options in Factory classes

* Add Reuters to Kim-CNN command line arguments

* Fix SST dataset path according to changes in Kim-CNN args

The dataset path in args.py was made to point at the dataset folder rather than dataset/SST folder. Hence SST folder was added to paths in the SST dataset class

* Add Reuters dataset class, and support in __main__

* Add Reuters dataset trainers and evaluators

* Remove debug print statement in reuters_evaluator

* Fix rounding bug in reuters_trainer and reuters_evaluator
 2018-10-02 21:09:51 -04:00
Victor Yang 10e4e56817 Add twitter url dataset (#145) 
* add prediction/qrel files dump option

* fix comment

* add twitter-url dataset, minor refactor

* fix minor error
 2018-09-08 11:16:23 +08: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
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
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