Skip to content

Commit

Permalink
Update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
HessTaha authored May 31, 2020
1 parent 4d10245 commit d0e8a00
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,16 @@ You will find here :

## Model training
The architecture is basic, BiLSTM with LayerNormalization.

The model has been trained for 3 epochs with Adam optimizer and Cyclical LR Schedueler

### run training

```shell
python main.py --data_path='../Data/IMDB Dataset.csv' --texts_col='review' --labels_col='sentiment' --n_classes=2 --batch_size=16 --batch_size_eval=64 --n_epochs=2 --cuda=1
python main.py --data_path='../Data/IMDB Dataset.csv'
--texts_col='review' --labels_col='sentiment'
--n_classes=2 --batch_size=16 --batch_size_eval=64
--n_epochs=2 --cuda=1
```

## APP
Expand Down

0 comments on commit d0e8a00

Please sign in to comment.