def train_cafe(): voca = load_voca() args.char_vocab_size = load_char_length() model = Manager(max_sequence=100, word_indice=voca, batch_size=args.batch_size, num_classes=3, vocab_size=1000, embedding_size=300, lstm_dim=1024) data = models.common.load_pickle("train_corpus.pickle") validate = models.common.load_pickle("dev_corpus") epochs = 30 model.train(epochs, data, validate)
def train_keep_cafe(): voca = load_voca() manager = Manager(max_sequence=100, word_indice=voca, batch_size=args.batch_size, num_classes=3, vocab_size=1000, embedding_size=300, lstm_dim=1024) # Dev acc=0.6576999819278717 loss=0.8433943867683411 data = models.common.load_pickle("train_corpus.pickle") validate = models.common.load_pickle("dev_corpus") manager.load("model-15340") manager.train(20, data, validate, True)