Skip to content

Commit 6dfc862

Browse files
author
wy
committed
use nonlinear classifier by default
1 parent 299f8ca commit 6dfc862

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

README.md

+8
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,11 @@ python preprocess_dataset.py --glove_file /path/to/your/glove --input_files /pat
1717
```bash
1818
python train.py --glove_file /path/to/your/glove --input_train_file_pattern "/path/to/save/tfrecords/train-?????-of-?????" --input_valid_file_pattern "/path/to/save/tfrecords/valid-?????-of-?????" --train_dir /path/to/save/checkpoints
1919
```
20+
21+
## Experiment result
22+
With the default settings in configurations.py, I obtained a dev accuracy of 83.72% in epoch 5,
23+
83.37% in epoch 10.
24+
25+
## Tips for fine-tuning
26+
Don't dropout too much. When the classifier and encoder dropout are both set to 0.5, both the
27+
train and the dev accuracy is decreased below 70%.

configuration.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def __init__(self):
1414
self.shuffle = True
1515
self.uniform_init_scale = 0.1
1616
self.num_classes = 3
17-
self.nonlinear_classifier = False
17+
self.nonlinear_classifier = True
1818
self.encoder_dropout = 0.0
1919
self.classifier_dropout = 0.0
2020

0 commit comments

Comments
 (0)