INNER CODE UNIT · Python

Bot

shibing624/dialogbot · dialogbot/bot.py:13

class Bot:
    def __init__(
            self,
            vocab_path=config.vocab_path,
            search_model=config.search_model,
            question_answer_path=config.question_answer_path,
            context_response_path=config.context_response_path,
            context=None,
            model_name_or_path="shibing624/gpt2-dialogbot-base-chinese",
            max_history_len=3,
            max_len=128,
            repetition_penalty=1.0,
            temperature=1.0,
            topk=8,
            topp=0.0,
            last_txt_len=100

    ):

View source record →