trqminh / snlp_final_proj

text clf

snlp_final_proj

Project logs

View project training logs here

Datasets and train models

this_repo/
  data/
    bbc_news/

  trained_models/
    bert_bbc_bestover50.pth
    distilbert_bbc_bestover50.pth
    bert_imdb_bestover50.pth
    distilbert_imdb_bestover50.pth
    ...

Download datasets and trained models from here. For aclImdb_v1 dataset, download the dataset from here

To train

e.g.

python train.py --model distilbert --dataset bbc --epochs 50

To test

e.g.

python test.py --model bert --dataset bbc --model_path trained_models/bert_bbc_bestover50.pth