Exemplo n.º 1
0
def do_shell(_):
    with tf.Session() as sess:
        saver.restore(sess, checkpoint_path)
        sentence = '请输入句子,要退出请输入:bye'
        logger.info(sentence)
        while sentence != 'bye':
            sentence = input("input> ")
            if sentence == 'bye':
                logger.info("准备退出...")
            elif sentence == '':
                logger.info("请不要打回车玩哦!")
            elif '__user__' not in sentence:
                logger.info("请输入正确的格式!提示:Question__user__Answer;退出请输入:bye")
            else:
                shell_gen = generate_batch_shell(sentence)
                source, source_seq_length, target, target_seq_length = next(
                    shell_gen)
                preds = sess.run(pred, feed_dict={encode_question: source, encode_answer: target,
                                                  question_sequence_length: source_seq_length, answer_sequence_length: target_seq_length})
                predict = preds[0]
                logger.info("输入原句:{}".format(sentence))
                if predict[0] == 1:
                    logger.info("输出结果:该句识别准确")
                else:
                    logger.info("输出结果:该句识别不准确")
        logger.info('谢谢使用,再见!')
Exemplo n.º 2
0
def do_shell(_):
    with tf.Session() as sess:
        saver.restore(sess, checkpoint_path)
        sentence = '请输入句子,要退出请输入:bye'
        logger.info(sentence)
        while sentence != 'bye':
            sentence = input("input> ")
            if sentence == 'bye':
                logger.info("准备退出...")
            elif sentence == '':
                logger.info("请不要打回车玩哦!")
            elif '\t' not in sentence:
                logger.info("请输入正确的格式!提示:Question\tAnswer;退出请输入:bye")
            else:
                shell_gen = generate_batch_shell(sentence)
                source, target = next(shell_gen)
                preds = sess.run(pred, feed_dict={encode_question: source, encode_answer: target})
                predict = preds[0]
                intention = label_dict[str(predict)]
                logger.info("输入原句:{}".format(sentence))
                logger.info("输出意图:{}".format(intention))
        logger.info('谢谢使用,再见!')