NLP/nlp_xiaojiang
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update args.py

Browse Source
This commit is contained in:
yongzhuo 2019-05-20 19:50:41 +08:00 committed by GitHub
parent 33362cd85b
commit cee989d35d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ClassificationText/bert/args.py
View File

@ -1,4 +1,4 @@
# bi-lstm
return_sequences = True return_sequences = True
use_cudnn_cell = True use_cudnn_cell = True
use_lstm = True use_lstm = True
@ -8,6 +8,7 @@ metrics = ['accuracy']
activation = 'softmax' activation = 'softmax'
optimizers = 'adam' optimizers = 'adam'
learning_rate = 1e-3 learning_rate = 1e-3
embedding_dim = 768
keep_prob = 0.5 keep_prob = 0.5
units = 256 units = 256
decay = 0.0 decay = 0.0
@ -15,9 +16,12 @@ label = 2
l2 = 0.32 l2 = 0.32
epochs = 100 epochs = 100
batch_size = 32 batch_size = 128
path_save_model = 'model_webank_tdt/bert_bi_lstm_pair.h5' # 'bert_bi_lstm_pair.h5' path_save_model = 'model_webank_tdt/bert_avt_cnn.h5' # 'bert_bi_lstm_pair.h5'
# text-cnn
filters = [3, 4, 5]
num_filters = 512