See tests/functional/test_classification.py for details on how to use.
python src/main.py \
--model 'roberta' \
--output-dir 'output' \
--data-path 'data/fake_news_sample.csv'-
Notifications
You must be signed in to change notification settings - Fork 0
raufer/text-classification-pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Text classification with fine-tune of a pretrained model
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published