diff options
author | TharinduDR <rhtdranasinghe@gmail.com> | 2021-06-03 15:18:09 +0300 |
---|---|---|
committer | TharinduDR <rhtdranasinghe@gmail.com> | 2021-06-03 15:18:09 +0300 |
commit | 66ae8f324989e2e910f7e7ddb953b819cbdea0a3 (patch) | |
tree | 34cad00c4b97f936a71473cb772d5bc2e4e322c9 | |
parent | 61eceb744ee9508af091bb4cc5afcc98be44d7cf (diff) |
057: Code Refactoring - Siamese Architectures
-rw-r--r-- | docs/models/word_level_pretrained.md | 28 |
1 files changed, 20 insertions, 8 deletions
diff --git a/docs/models/word_level_pretrained.md b/docs/models/word_level_pretrained.md index 255478d..b173c6c 100644 --- a/docs/models/word_level_pretrained.md +++ b/docs/models/word_level_pretrained.md @@ -1,21 +1,33 @@ # Word Level Pre-trained Models -We have released several pre-trained TransQuest models on word-level quality estimation. We will be keep releasing new models. So please keep in touch. +We have released several pre-trained TransQuest models on word-level quality estimation and they are available on [Hugging Face Model Hub](). We will be keep releasing new models. So please keep in touch. ### Available Models -| Language Pair | NMT/SMT | Domain | Algorithm | Model Link | +| Language Pair | NMT/SMT | Domain | Algorithm | Model Name and Link to HUgging Face | |:---------------:|:--------------:|:------------:|:--------------------:|:------------------------------------:| -| English-German | NMT | IT | MicroTransQuest | [model.zip](http://bit.ly/2QbsXN9) | +| English-German | NMT | IT | MicroTransQuest | [TransQuest/microtransquest-en_de-it-nmt](https://huggingface.co/TransQuest/microtransquest-en_de-it-nmt) | | | NMT | Wiki | MicroTransQuest | | -| | SMT | IT | MicroTransQuest | [model.zip](http://bit.ly/38ZgVwZ) | -| English-Latvian | NMT | Life Sciences| MicroTransQuest | [model.zip](http://bit.ly/3eR4bfI) | -| | SMT | Life Sciences| MicroTransQuest | [model.zip](http://bit.ly/3cMrepn) | -| English-Czech | SMT | IT | MicroTransQuest | [model.zip](http://bit.ly/3r2bjZc) | -| German-English | SMT | Life Sciences| MicroTransQuest | [model.zip](http://bit.ly/2NweqdY) | +| | SMT | IT | MicroTransQuest | [TransQuest/microtransquest-en_de-it-smt](https://huggingface.co/TransQuest/microtransquest-en_de-it-smt) | +| English-Latvian | NMT | Life Sciences| MicroTransQuest | [TransQuest/microtransquest-en_lv-pharmaceutical-nmt](https://huggingface.co/TransQuest/microtransquest-en_lv-pharmaceutical-nmt)| +| | SMT | Life Sciences| MicroTransQuest | [TransQuest/microtransquest-en_lv-pharmaceutical-smt](https://huggingface.co/TransQuest/microtransquest-en_lv-pharmaceutical-smt) | +| English-Czech | SMT | IT | MicroTransQuest | [TransQuest/microtransquest-en_cs-it-smt ](https://huggingface.co/TransQuest/microtransquest-en_cs-it-smt) | +| German-English | SMT | Life Sciences| MicroTransQuest | [TransQuest/microtransquest-de_en-pharmaceutical-smt](https://huggingface.co/TransQuest/microtransquest-de_en-pharmaceutical-smt) | | English-Chinese | NMT | Wikipedia | MicroTransQuest | | | English-Russian | NMT | IT | MicroTransQuest | | | \*-\* | Any | Any | MicroTransQuest | | +You can load any of the above models with the below code. The full notebook is available [here.](https://colab.research.google.com/drive/1fslfFoQnspdv2Do5hfwwkbEwow-CDIxD?usp=sharing) + +```python +from transquest.algo.word_level.microtransquest.run_model import MicroTransQuestModel +import torch + +model = MicroTransQuestModel("xlmroberta", "TransQuest/microtransquest-en_lv-pharmaceutical-nmt", labels=["OK", "BAD"], use_cuda=torch.cuda.is_available()) +``` + + + + !!! note \* denotes any language. (\*-\* means any language to any language)
\ No newline at end of file |