# AistrighNLP
<p align="center">
<img src="https://github.com/JustCunn/AistrighNLP/blob/master/images/github_aistrigh.png" width="150">
<br />
<br />
<a href="https://github.com/JustCunn/AistrighNLP/blob/master/LICENSE"><img alt="MIT License" src="https://img.shields.io/github/license/JustCunn/AistrighNLP?style=for-the-badge" /></a>
<a href="https://github.com/JustCunn/AistrighNLP/releases"><img alt="Release" src="https://img.shields.io/github/release/JustCunn/AistrighNLP" /></a>
</p>
AistrighNLP is a collection of tools and models used for Aistrigh, the BT Young Scientist 2021 project. Our aim is to bring Irish into the modern era with NLP tools to give it parity with English.
The tools included are based around the work in [Neural Models for Predicting Celtic Mutations](https://www.aclweb.org/anthology/2020.sltu-1.1.pdf) (Scannell, 2020). Included is all the tools needed to create a demutated Irish corpus, which can be used in all sorts of NLP tasks, and a model to reinsert them. For the Python API docs visit [AistrighNLP Python API](https://github.com/JustCunn/AistrighNLP/blob/master/python.md)
Installing the Package
---------------------------
AistrighNLP can be downloaded using `pip`
```
pip install aistrigh-nlp
```
Lowercasing
---------------------------
When lowercasing either Irish or Scots Gaelic for prediciting mutations, you must be aware of special cases outlined in the paper above. Our lowercaser handles that
```
aistrigh-nlp lowercase -i input.txt -o output.txt
```
Removing mutations
-----------------------
To remove mutations from an entire dataset for use for NLP tasks (like Machine Translation) use `demutate-corpus`. `-l`/`--lang` must take ISO 639 language codes like 'ga'.
```
aistrigh-nlp demutate-corpus -i input.txt -o output.txt -l ga
```
To remove mutations with a 'window' on either side to train a neural network, use `demutate-window`, with `-w` set to your desired window length on each side
```
aistrigh-nlp demutate-window -i input.txt -o output.csv -l ga -w 16
```
Predicting the mutations
-------------------------
To predict mutations on each word, use `predict-mutations`. As of right now, it's only compatible with PyTorch+Torchtext models but we are working on expanding to TensorFlow and Keras. You'll need your vocab, labels and model checkpoint in the same folder (`-d`/`--data`). We provide default models to be used [here](https://sites.google.com/view/aistrighnlp/cáipéisiocht-documentation).
```
aistrigh-nlp predict-mutations -i input.txt -o output.txt -w 16 -d nn_100k
```
Applying the predicted mutations
-----------------------------------
To apply the mutations predicted by `predict-mutations`, use `apply-mutations`.
```
aistrigh-nlp apply-mutations -i input.txt -o output.txt -l ga
```
Scoring Celtic NMT models using Standard and Demutated BLEU
-------------------------------------------------------------
To score NMT models using both these metrics run;
```
aistrigh-nlp bleu -r reference.txt -p predictions.txt -l ga
```
If you're scoring a demutated NMT model and haven't reapplied mutations, pass your demutated reference (`-d`) and predictions, and the original reference (`-r`).
```
aistrigh-nlp bleu -d demutated_reference.txt -r reference.txt -p predictions.txt -l ga
```
NOTE
--------------------
AistrighNLP uses PyTorch `Traces` to save the full computational graphs as checkpoints. This way, the model architecture need not be declared into hard-coded scripts. See this [StackOverflow Thread](https://stackoverflow.com/questions/59287728/saving-pytorch-model-with-no-access-to-model-class-code) for instructions to save a traced checkpoint.