Skip to content

Update config for text-classification #325

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Sep 19, 2023
Merged

Update config for text-classification #325

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
6f9b4fe
Update config for text-classification
puhuk Sep 19, 2023
857a6be
Update config.yaml
puhuk Sep 19, 2023
59195f8
Update config.yaml
puhuk Sep 19, 2023
b8d57b8
Update config.yaml
puhuk Sep 19, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 9 additions & 9 deletions src/templates/template-text-classification/config.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
#::= from_template_common ::#
model: bert-base-uncased
model_dir: /tmp/model
tokenizer_dir: /tmp/tokenizer
num_classes: 1
drop_out: .3
n_fc: 768
weight_decay: 0.01
num_warmup_epochs: 0
max_length: 256
model: bert-base-uncased # model for text classification
model_dir: /tmp/model # path of model directory
tokenizer_dir: /tmp/tokenizer # path of directory containing tokenizer
num_classes: 1 # number of classes
drop_out: .3 # drop out rate
n_fc: 768 # number of fully-connected layers
weight_decay: 0.01 # parameter for weight decay, default, 1e-4.
num_warmup_epochs: 0 # number of warm-up epochs before learning rate decay
max_length: 256 # maximum length of texts
lr: 0.00005