def test_intent_classification():
nludatafile = "/media/nvidia/ssd/catkin_ws/src/chu_bot_source/chubot/data/nlu_greet.json"
chubot = ChuBotBrain("greet_en", language='en')
chubot.load_data(nludatafile)
meta = chubot.train_intent_classification()
print(meta)
inmessage = "sad but great. want a dog picture"
intent_probs = chubot.predict_intent(inmessage)
print(intent_probs)
def test_intent_train(name):
chubot = ChuBotBrain(name, language='vi')
chubot.load_data("data/train.json")
# chubot.load_data("data/vi_nlu_ask_way.json")
meta = chubot.train_intent_classification()
test_link = "data/test.txt"
with open(test_link,'r',encoding='utf=8') as f:
rows = f.readlines()
intent_list_test = []
intent_list_result = []
data_list_test = []
for row in rows:
parts = row.split(',')
intent_list_test.append(encode_intent(parts[0]))
data_list_test.append(parts[1])
for data in data_list_test:
responses = chubot.predict_intent(data)
(prob, intent) = responses[0]
intent_list_result.append(encode_intent(intent))
print("accuracy",accuracy_score(intent_list_test,intent_list_result))