def test_run(self):
main.run(data_file=self.path_input_file,
is_train=True,
batchsize=32,
model_name="test_bilstm-cnn-crf",
optimizer="adam",
save_name="test_bilstm-cnn-crf",
save_dir=mkdtemp(),
gpu=-1,
dev_file=self.path_input_file,
test_file=self.path_input_file,
delimiter=" ",
input_idx="0",
output_idx="-1",
n_word_emb=100,
n_hidden=200,
n_layer=1,
n_char_emb=30,
n_char_hidden=30,
n_add_feature_emb=100,
use_cudnn=1,
init_lr=0.001,
model_filename=None,
max_iter=50)
def predict(input_file, **args):
main.run(input_file, is_train=False, **args)
def train(train_file, **args):
# load input_file
main.run(train_file, is_train=True, **args)