def create_classifier(n_vocab, doc_length, wv_size, filter_sizes, hidden_units, output_channel, initialW, non_static, batch_size, epoch, gpu):
model = NNModel(n_vocab=n_vocab,
doc_length=doc_length,
wv_size=wv_size,
filter_sizes=filter_sizes,
hidden_units=hidden_units,
output_channel=output_channel,
initialW=initialW,
non_static=non_static)
# optimizer = optimizers.Adam()
optimizer = optimizers.AdaDelta()
return (model, ChainerEstimator(model=SoftmaxCrossEntropyClassifier(model),
optimizer=optimizer,
batch_size=batch_size,
device=gpu,
stop_trigger=(epoch, 'epoch')))
评论列表
文章目录