Skip to content

Text Classification Repository that is able to train on uncertainty method.

Notifications You must be signed in to change notification settings

counterfake/text_category_classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

  1. Generate Preprocessed Data
python3 generate_data.py \
  --data_path raw_data.csv \
  --data_name category_classificaton_v1
  1. Train Model
python3 train.py \
  --model_path base_model \
  --name model_v1 \
  --data_path category_classificaton_v1.csv
  1. Test Model
python3 test.py \
  --test_path /your/test_data/path \
  --model_directory /your/model_folder/path

About

Text Classification Repository that is able to train on uncertainty method.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%