如何测试RASA模型?

时间:2019-05-22 09:29:24

标签: python nlp chatbot rasa-nlu rasa-core

我正在尝试使用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 个答案:

答案 0 :(得分:1)

模块已重命名。 请导入

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)