wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity
143 Commits 6 Branches 0 Tags
master
Commit Graph
9 Commits
Author SHA1 Message Date
Ralph Tang 3855254870 Tidy up some code (#174) 2019-02-06 13:36:48 -05:00
Michael Tu 0c3a91c443 Check in MP-CNN Lite Model (#108) 
* Add MP-CNN Lite model

* MP-CNN Lite bug fixes
 2018-05-25 00:15:50 -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 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
Michael Tu 09b3a790a2 Use torchtext for MP-CNN (#76) 
* Add SICK torchtext Dataset

* SICK dataset - torchtext postprocess into class probs

* Update model, driver, trainer, evaluator for SICK for torchtext

* MP-CNN: Fix bugs that prevent SICK from running on gpu 0

* MP-CNN: make SICK dataset w/ torchtext GPU-agnostic

* MP-CNN: support sparse features / idf overlap with torchtext

* Add MSRVID dataset with torchtext and update MP-CNN code to use it

* MP-CNN: Make torchtext deterministic by setting python random seed

* SICK and MSRVID datasets - add pair id for debug and build test vocab

* MP-CNN: Update readme to address potential module not found error

* MP-CNN: address review comments, can run on cpu
 2017-11-01 12:13:30 -04:00
Michael Tu aba4dd7e89 MP-CNN: Add Additional Features (#51) 
* MP-CNN: support external features

* MP-CNN: calculate overlap in same way as SM-model

* MP-CNN: properly calculate idf overlap
 2017-09-21 22:43:31 -04:00
Michael Tu a2904efe5a MP-CNN Bugfixes and Improvements (#50) 
* MP-CNN: use consistent unknown vector

* MP-CNN: Make optimizer, patience, etc.. configurable

* MP-CNN: bug fixes

* MP-CNN: update README

* MP-CNN: remove unused import
 2017-09-21 15:53:54 -04:00
Michael Tu 449d715ab0 MP-CNN PyTorch Re-implementation (#37) 
Re-implementation of MP-CNN in PyTorch.
 2017-08-04 19:37:34 -04:00