Как проверить модель RASA? - PullRequest
0 голосов
/ 22 мая 2019

Я пытаюсь написать своего собственного чата с фреймворком RASA.
Сейчас я просто играю с этим, и у меня есть следующий фрагмент кода для учебных целей.

from rasa.nlu.training_data import load_data
from rasa.nlu.config import RasaNLUModelConfig
from rasa.nlu.model import Trainer
from rasa.nlu import config
training_data = load_data("./data/nlu.md")
trainer = Trainer(config.load("config.yml"))
interpreter = trainer.train(training_data)
model_directory = trainer.persist("./models/nlu",fixed_model_name="current")

Теперь я прочитал, что если я хочу проверить это, я должен сделать что-то вроде этого.

from rasa.nlu.evaluate import run_evaluation
run_evaluation("nlu.md", model_directory)

Но этот код больше не доступен ни в rasa.nlu.evaluate, ни в rasa.nlu.test!
Каков способ тестирования модели RASA?

1 Ответ

1 голос
/ 24 мая 2019

Модуль был переименован. Пожалуйста, импортируйте

from rasa.nlu.test import run_evaluation

В качестве альтернативы вы теперь также можете

from rasa.nlu import test
test_result = test(path_to_test_data, unpacked_model)

intent_evaluation_report = test_result["intent_evaluation"]["report"]
print(intent_evaluation_report)
...