def __init__(self, vocab_size, embed_size, hidden_size): super(EncoderDecoder, self).__init__( enc=Encoder(vocab_size, embed_size, hidden_size), dec=Decoder(vocab_size, embed_size, hidden_size), ) self.vocab_size = vocab_size self.embed_size = embed_size self.hidden_size = hidden_size self.common_function = CommonFunction()
def __init__(self, parameter_dict): self.parameter_dict = parameter_dict self.source = parameter_dict["source"] self.target = parameter_dict["target"] self.test_source = parameter_dict["test_source"] self.test_target = parameter_dict["test_target"] self.vocab = parameter_dict["vocab"] self.embed = parameter_dict["embed"] self.hidden = parameter_dict["hidden"] self.epoch = parameter_dict["epoch"] self.minibatch = parameter_dict["minibatch"] self.generation_limit = parameter_dict["generation_limit"] self.word2vec = parameter_dict["word2vec"] self.word2vecFlag = parameter_dict["word2vecFlag"] self.common_function = CommonFunction() self.model = "../ChainerDialogue"