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))