File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 30
30
DROPRATE ,
31
31
REGULARIZATION_CONSTANT ,
32
32
BILOU_FLAG ,
33
- SHARE_HIDDEN_LAYERS ,
34
- UNIDIRECTIONAL_ENCODER ,
35
33
)
36
34
from rasa .utils .common import raise_warning
37
35
from rasa .utils .tensorflow .models import RasaModel
@@ -72,13 +70,13 @@ class CRFEntityExtractor(DIETClassifier):
72
70
# nn architecture
73
71
# sizes of hidden layers before the embedding layer for input words
74
72
# the number of hidden layers is thus equal to the length of this list
75
- HIDDEN_LAYERS_SIZES : {TEXT : [256 , 128 ]},
73
+ HIDDEN_LAYERS_SIZES : {TEXT : []},
76
74
# training parameters
77
75
# initial and final batch sizes - batch size will be
78
76
# linearly increased for each epoch
79
77
BATCH_SIZES : [64 , 256 ],
80
78
# how to create batches
81
- BATCH_STRATEGY : "balanced " , # string 'sequence' or 'balanced'
79
+ BATCH_STRATEGY : "sequence " , # string 'sequence' or 'balanced'
82
80
# number of epochs
83
81
EPOCHS : 300 ,
84
82
# set random seed to any int to get reproducible results
@@ -103,7 +101,7 @@ class CRFEntityExtractor(DIETClassifier):
103
101
# BILOU_flag determines whether to use BILOU tagging or not.
104
102
# More rigorous however requires more examples per entity
105
103
# rule of thumb: use only if more than 100 egs. per entity
106
- BILOU_FLAG : False ,
104
+ BILOU_FLAG : True ,
107
105
}
108
106
# end default properties (DOC MARKER - don't remove)
109
107
You can’t perform that action at this time.
0 commit comments