Extract model predictions

cpllab / lm-zoo

Easy black-box access to state-of-the-art language models

https://cpllab.github.io/lm-zoo/

MIT License

14 stars 6 forks source link

Extract model predictions #39

Closed hans closed 4 years ago

hans commented 4 years ago

Closes #26 . This implements prediction extraction for models

GRNN
JRNN
all transformers models

and leaves unimplemented

ngram (requires breaking open SRILM code to support)
RNNG (requires breaking open RNNG dynet code to support)
ordered-neurons (haven't started)

I also added registry+CLI features for stating supported features, with the idea that not all models will support all features in each release. This is yet to be implemented in the lm-zoo API or CLI frontend; see #33 .

hans commented 4 years ago

This is a backward-compatible release, so I'll build and distribute all images now before merging.

hans commented 4 years ago

All good. CI failures are not relevant.