-
Notifications
You must be signed in to change notification settings - Fork 2
/
configs.py
78 lines (72 loc) · 4.63 KB
/
configs.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import tensorflow as tf
## Hyperparameter for CNN
# tf.app.flags.DEFINE_integer('batch_size', 32, 'batch size')
# tf.app.flags.DEFINE_integer('epoch', 5, 'max epoch')
# tf.app.flags.DEFINE_integer('train_step', 800, 'train_step')
# tf.app.flags.DEFINE_integer('hidden_dim', 100, 'model dim')
# tf.app.flags.DEFINE_integer('vocab_size', 16581, 'model dim')
# tf.app.flags.DEFINE_integer('emb_dim', 300, 'embedding dim')
# tf.app.flags.DEFINE_integer('n_label', 5, 'embedding dim')
# tf.app.flags.DEFINE_list('filter_size', [3,4,5], 'filter size')
# tf.app.flags.DEFINE_integer('num_filters', 100, 'number of filters')
# tf.app.flags.DEFINE_float('learning_rate', 0.0005, 'learning rate')
# tf.app.flags.DEFINE_float('dropout_rate', 0.5, 'dropout rate')
# tf.app.flags.DEFINE_string('data_path', './data/', 'data path')
# tf.app.flags.DEFINE_string('output_path', './output/', 'output path')
# tf.app.flags.DEFINE_string('ckpt_path', './output/ckpt/', 'checkpoint path')
# tf.app.flags.DEFINE_string('best_ckpt_path', './output/best_ckpt/', 'best_checkpoint path')
# tf.app.flags.DEFINE_string('train_data', 'stsa_fine_train.txt', 'training data file name')
# tf.app.flags.DEFINE_string('dev_data', 'stsa_fine_dev.txt', 'dev data file name')
# tf.app.flags.DEFINE_string('test_data', 'stsa_fine_test.txt', 'test data file name')
# tf.app.flags.DEFINE_integer('max_seq_length', 40, 'max_seq_length')
# tf.app.flags.DEFINE_bool('train', True, 'run trainig')
# tf.app.flags.DEFINE_string('model', 'CNN', 'select a training model')
## Hyperparameter for LSTM
tf.app.flags.DEFINE_integer('batch_size', 32, 'batch size')
tf.app.flags.DEFINE_integer('epoch', 10, 'max epoch')
tf.app.flags.DEFINE_integer('train_step', 1000, 'train_step')
tf.app.flags.DEFINE_integer('hidden_dim', 300, 'model dim')
tf.app.flags.DEFINE_integer('vocab_size', 16581, 'model dim')
tf.app.flags.DEFINE_integer('emb_dim', 300, 'embedding dim')
tf.app.flags.DEFINE_integer('n_label', 5, 'embedding dim')
tf.app.flags.DEFINE_integer('d_a_size', 350, 'attention_size')
tf.app.flags.DEFINE_integer('r_size', 30, 'attention_size_2')
tf.app.flags.DEFINE_float('p_coef', 1.0, 'penalty')
tf.app.flags.DEFINE_list('filter_size', [3,4,5], 'filter size')
tf.app.flags.DEFINE_integer('num_filters', 100, 'number of filters')
tf.app.flags.DEFINE_float('learning_rate', 1e-3, 'learning rate')
tf.app.flags.DEFINE_float('dropout_rate', 0.5, 'dropout rate')
tf.app.flags.DEFINE_string('data_path', './data/', 'data path')
tf.app.flags.DEFINE_string('output_path', './output/', 'output path')
tf.app.flags.DEFINE_string('ckpt_path', './output/ckpt/', 'checkpoint path')
tf.app.flags.DEFINE_string('best_ckpt_path', './output/best_ckpt/', 'best_checkpoint path')
tf.app.flags.DEFINE_string('train_data', 'stsa_fine_train.txt', 'training data file name')
tf.app.flags.DEFINE_string('dev_data', 'stsa_fine_dev.txt', 'dev data file name')
tf.app.flags.DEFINE_string('test_data', 'stsa_fine_test.txt', 'test data file name')
tf.app.flags.DEFINE_integer('max_seq_length', 52, 'max_seq_length')
tf.app.flags.DEFINE_bool('train', True, 'run trainig')
tf.app.flags.DEFINE_string('model', 'BILSTMCNN', 'select a training model')
## for binary
# tf.app.flags.DEFINE_integer('batch_size', 32, 'batch size')
# tf.app.flags.DEFINE_integer('epoch', 5, 'epoch')
# tf.app.flags.DEFINE_integer('train_step', 600, 'train_step')
# tf.app.flags.DEFINE_integer('hidden_dim', 128, 'model dim')
# tf.app.flags.DEFINE_integer('vocab_size', 14830, 'model dim')
# tf.app.flags.DEFINE_integer('emb_dim', 256, 'embedding dim')
# tf.app.flags.DEFINE_integer('n_label', 5, 'embedding dim')
# tf.app.flags.DEFINE_list('filter_size', [3,4,5], 'filter size')
# tf.app.flags.DEFINE_integer('num_filters', 100, 'number of filters')
# tf.app.flags.DEFINE_float('learning_rate', 0.0005, 'learning rate')
# tf.app.flags.DEFINE_float('dropout_rate', 0.3, 'dropout rate')
# tf.app.flags.DEFINE_string('data_path', './data/', 'data path')
# tf.app.flags.DEFINE_string('output_path', './output/', 'output path')
# tf.app.flags.DEFINE_string('ckpt_path', './output/ckpt/', 'checkpoint path')
# tf.app.flags.DEFINE_string('best_ckpt_path', './output/best_ckpt/', 'best_checkpoint path')
# tf.app.flags.DEFINE_string('train_data', 'stsa_binary_train.txt', 'training data file name')
# tf.app.flags.DEFINE_string('dev_data', 'stsa_binary_dev.txt', 'dev data file name')
# tf.app.flags.DEFINE_string('test_data', 'stsa_binary_test.txt', 'test data file name')
# tf.app.flags.DEFINE_integer('max_seq_length', 20, 'max_seq_length')
# tf.app.flags.DEFINE_bool('train', True, 'run trainig')
# tf.app.flags.DEFINE_string('model', 'LSTM', 'select a training model')
tf.app.flags.DEFINE_string('f', '', 'kernel')
DEFINES = tf.app.flags.FLAGS