Skip to content

Commit cb7d195

Browse files
committed
added random seed and number of epochs to crf extractor test
1 parent 62c9b5f commit cb7d195

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/nlu/conftest.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from rasa.nlu import config, train
77
from rasa.nlu.components import ComponentBuilder
8-
from rasa.utils.tensorflow.constants import EPOCHS
8+
from rasa.utils.tensorflow.constants import EPOCHS, RANDOM_SEED
99

1010
CONFIG_DEFAULTS_PATH = "sample_configs/config_defaults.yml"
1111

@@ -42,7 +42,9 @@ def ner_crf_pos_feature_config():
4242
["low", "title", "upper", "pos", "pos2"],
4343
["low", "suffix3", "suffix2", "upper", "title", "digit", "pos", "pos2"],
4444
["low", "title", "upper", "pos", "pos2"],
45-
]
45+
],
46+
EPOCHS: 100,
47+
RANDOM_SEED: 2020,
4648
}
4749

4850

0 commit comments

Comments
 (0)