Commit Graph
22 Commits
Author SHA1 Message Date
Michael Tu 5bf33bf8ea Update README to use Castor-models and Instructions for Internal Users (#113)
* Update instructions to use Castor-models
* Consolidate requirements.txt
* Refine README with convenience scripts
* Update internal instructions
* MP-CNN working dir minor edit
2018-05-25 18:14:10 -04:00
Michael Tu f7a0167b81 Migrate to from GitHub castorini/data to uWaterloo Castor-data (#103)
* Refactor main README
* Update Anserini Dependency docs
* Update idf baseline and Kim CNN docs to use Castor-data
* Update remaining READMEs to reference Castor-data
* Change default path from data to Castor-data
* Fix wrong order of embeddings path
2018-05-23 16:16:17 -04:00
Michael Tu eee160ea41 Use view instead of unsqueeze since ONNX v1.0 doesn't support it (#94) 2017-12-08 15:04:29 -05:00
rosequ 85f35bb994 connected the pipeline (#65) (#93)
* connected the pipeline

* minor changes to api code
2017-12-07 20:04:18 -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 a363e3d256 Add flag to support onnx (#91) 2017-12-06 09:55:02 -05:00
rosequ 0a3ce7015c Update readme (#90)
* updated readme for easier replication

* updated cd changes

* link change
2017-12-05 21:58:55 -05:00
rosequ 28a198f33c Update bridge (#87) (#89)
* initial commit of the updated bridge

* moved bridge file to the root

* after CR1

* after CR2
2017-12-05 12:23:08 -05:00
Ralph Tang e7ca33de28 Add convolutional RNN for sentence classification (#57)
* Add SST data preprocessing

* Add ConvRNN model

* Add LR scheduler

* Add grid search on hyperparameters

* Add random search

* Add CLI options

* Add usage to README.md

* Refactor code

* Fix randomized search parameters

* Update README.md with results

* Use Dataset and DataLoader
2017-10-05 14:14:31 -04:00
Matt Crane ed4dba2497 Typo fix (#54) 2017-09-22 11:30:26 +09:00
Matt Crane a601722768 Option to disable the CuDNN backend (#52)
Some of the kernels in this backend are known to be non-deterministic. It's also not clear to me whether it's enabled in a default install or not, enabled is set to True by default, but I think ultimately it's a runtime choice.
2017-09-22 11:19:30 +09:00
Michael Tu 245b032545 Set gpu seed for SM model (#46)
Small fix to set GPU seed for SM model
2017-09-12 14:28:18 -04:00
Matt Crane a3294339fb Add seed and thread arguments (#43)
So that any experiments can be controlled from the command line, I've set the defaults to what they were hardcoded (for the seed) and num_threads in kim_cnn to be consistent with the default in sm_cnn.
2017-09-01 09:30:01 -04:00
Matt Crane b4dc87d46f Explicitly open files with utf-8 encoding (#35)
Same issue as castorini/data#19
2017-08-29 14:20:18 -04:00
Michael Tu a6fc10818f Fix SM Model Internal Reproducibility Bug (#34)
* Fix SM Model reproducibility bug

vocab is in different order every time, causing unseen words to use
different random states

* Make requirements.txt usable from conda and pip

The existing torch requirement does not work with conda or pip.
Also upgrade pytorch version while we are at it.
2017-07-20 06:52:26 +08:00
Michael Tu a0755e6aa3 SM Model Jupyter Notebook Tutorial (#32)
Notebook tutorial for SM CNN.
2017-07-04 18:28:17 -04:00
Salman Mohammed 4c081645a7 clean up the relation prediction model for Simple QA - Ferhan's paper (#28)
+ cleaned up the model code for the simple qa directory
+ created vocab objects for pre-loading word embeddings easily
2017-06-19 14:24:27 -04:00
Matt Crane 945b1fa6c0 Fix vocab caching issue (#29)
With the line as-was the vocab cache was stored as b'the' rather than the, meaning that word2vec wasn't found for terms causing massive performance loss (AP 0.71 cf 0.77).
2017-06-19 13:02:36 -04:00
Michael Tu 43dd6fdb1f GPU Support for SM Model (#26)
Add code to using GPU for the SM Model (#25). To use the GPU if one is available, add the --cuda optional parameter when calling main.py.
2017-06-02 08:47:13 -04:00
Gaurav Baruah 92789cb9f5 E2e sweep (#24)
Now ensuring that the bridge process raw candidate sentences fetched from the index, exactly as was done for the best performing SM model.
2017-05-29 19:47:42 -04:00
gauravbaruah 9bd5b7bb2a Corpus idf (#23)
as part of sourcing-IDF-from-index and e2e experiments.
2017-05-07 22:28:35 -04:00
rosequ 212aa6fb93 renaming sm_model; faster bridge (#18) (#22)
+ renamed sm_model
+ faster bridge by obtaining the IDF scores of a term directly from the Java server
2017-05-03 17:24:47 -04:00