Skip to content

simbirsoft/cce-loss-text

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Experiment on applying complement cross-entropy loss to text classification problem

Steps to reproduce

  1. Install dependencies:
pip install -r requirements.txt
  1. Donwload and extract data from Kaggle

  2. Preprocess the data:

python preprocess_data.py <path-to-train.tsv> <path-to-dataset-folder>
  1. Run experiment with standard cross-entropy loss:
python run_experiments.py <path-to-dataset-folder>

Add --gpu to train on GPU

  1. Run experiment with complement cross-entropy loss:
python run_experiments.py <path-to-dataset-folder> --use_cce
  1. Accuracy scores will appearn on screen. More details will be available in Tensorboard:
tensorboard --logdir=./runs

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages