File size: 1,002 Bytes
54f5ec9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
---
language: en
tags:
- grammar
- spell
- correction
---

# Spellcheck Model

## Installation
1. Install Python 3.8 or higher and Docker;
2. Clone the repo;
3. Create a virtual environment: `python3 -m venv venv`;
4. Activate the virtual environment: `source venv/bin/activate`;
5. Install dependencies: `pip install -r requirements.txt`;

#

## Usage
### Training
There's a `data` directory with two files:
1. `eval.csv`: evaluation dataset;
2. `train.csv`: training dataset;
Both are in the same format:
```txt
input,target
"grammar: som sentense","some sentence"
"grammar: anoder centence","another sentence"
...
```

You can edit them to use your own dataset.

Now if you want to fine-tune the existing model here, just run the following commands:
```bash
docker build -t spellcheck_train .
docker run -v $(pwd):/app spellcheck_train
```
Also make sure you have docker running on your machine.
### Testing
If you want to test your model, just run the following command:
```bash
python test.py
```