The evaluation script evaluate.py allows you to test your models performance for intent classification and entity recognition. You invoke this script supplying test data, model, and config file arguments:
python -m rasa_nlu.evaluate \ --data data/examples/rasa/demo-rasa.json \ --model projects/default/model_20180323-145833
Where model specifies the model to evaluate on the test data specified with data.
If you would like to evaluate your pipeline using cross-validation, you can run the evaluation script with the mode crossvalidation flag. This gives you an estimate of how accurately a predictive model will perform in practice. Note that you cannot specify a model in this mode, as a new model will be trained on part of the data for every crossvalidation loop. An example invocation of your script would be:
python -m rasa_nlu.evaluate \ --data data/examples/rasa/demo-rasa.json \ --config sample_configs/config_spacy.yml \ --mode crossvalidation
The evaluation script will log precision, recall, and f1 measure for each intent and once summarized for all. Furthermore, it creates a confusion matrix for you to see which intents are mistaken for which others.
For each entity extractor, the evaluation script logs its performance per entity type in your training data.
So if you use
ner_duckling in your pipeline, it will log two evaluation tables
containing recall, precision, and f1 measure for each entity type.
In the case
ner_duckling we actually run the evaluation for each defined
duckling dimension. If you use the
ordinal dimensions, you would
get two evaluation tables: one for
ner_duckling (Time) and one for
ner_synonyms does not create an evaluation table, because it only changes the value of the found
entities and does not find entity boundaries itself.
Finally, keep in mind that entity types in your testing data have to match the output
of the extraction components. This is particularly important for
ner_duckling, because it is not
fitted to your training data.
To evaluate entity extraction we apply a simple tag-based approach. We don’t consider BILOU tags, but only the entity type tags on a per token basis. For location entity like “near Alexanderplatz” we expect the labels “LOC” “LOC” instead of the BILOU-based “B-LOC” “L-LOC”. Our approach is more lenient when it comes to evaluation, as it rewards partial extraction and does not punish the splitting of entities. For example, the given the aforementioned entity “near Alexanderplatz” and a system that extracts “Alexanderplatz”, this reward the extraction of “Alexanderplatz” and punish the missed out word “near”. The BILOU-based approach, however, would label this as a complete failure since it expects Alexanderplatz to be labeled as a last token in an entity (L-LOC) instead of a single token entity (U-LOC). Also note, a splitted extraction of “near” and “Alexanderplatz” would get full scores on our approach and zero on the BILOU-based one.
Here’s a comparison between both different scoring mechanisms for the phrase “near Alexanderplatz tonight”:
|extracted||Simple tags (score)||BILOU tags (score)|
|[near Alexanderplatz](loc) [tonight](time)||loc loc time (3)||B-loc L-loc U-time (3)|
|[near](loc) [Alexanderplatz](loc) [tonight](time)||loc loc time (3)||U-loc U-loc U-time (1)|
|near [Alexanderplatz](loc) [tonight](time)||O loc time (2)||O U-loc U-time (1)|
|[near](loc) Alexanderplatz [tonight](time)||loc O time (2)||U-loc O U-time (1)|
|[near Alexanderplatz tonight](loc)||loc loc loc (2)||B-loc I-loc L-loc (1)|
There are a number of parameters you can pass to the evaluation script
$ python -m rasa_nlu.evaluate --help
Here is a quick overview:
usage: evaluate.py [-h] -d DATA [--mode MODE] [-c CONFIG] [-m MODEL] [-f FOLDS] [--debug] [-v] evaluate a Rasa NLU pipeline with cross validation or on external data optional arguments: -h, --help show this help message and exit -d DATA, --data DATA file containing training/evaluation data --mode MODE evaluation|crossvalidation (evaluate pretrained model or train model by crossvalidation) -c CONFIG, --config CONFIG model configurion file (crossvalidation only) -m MODEL, --model MODEL path to model (evaluation only) -f FOLDS, --folds FOLDS number of CV folds (crossvalidation only) --debug Print lots of debugging statements. Sets logging level to DEBUG -v, --verbose Be verbose. Sets logging level to INFO