def test_model(configs):
"""测试模型
"""
path_test = configs['data_params']['path_test'] if 'path_test' in configs[
'data_params'] else None
# init model
encoder, ptr_model, decoder = load_model(configs)
# init test data
data_test = load_data(path_test)
data_iter_test = init_test_data(configs)
# init infer
if 'path_test_result' not in configs['data_params'] or \
not configs['data_params']['path_test_result']:
path_result = configs['data_params']['path_test'] + '.result'
else:
path_result = configs['data_params']['path_test_result']
# label to id dict
path_pkl = os.path.join(configs['data_params']['alphabet_params']['path'],
'label.pkl')
label2id_dict = read_bin(path_pkl)
infer = Inference(encoder=encoder,
ptr_model=ptr_model,
decoder=decoder,
data_iter=data_iter_test,
data_raw=data_test,
path_result=path_result,
label2id_dict=label2id_dict)
# do infer
infer.infer2file()
def start_test(train_iter, dev_iter, test_iter, model, alphabet, config):
config.logger.info("\nTesting Start......")
data, path_source, path_result = load_test_data(train_iter, dev_iter,
test_iter, config)
infer = Inference(model=model,
data=data,
path_source=path_source,
path_result=path_result,
alphabet=alphabet,
use_crf=config.use_crf,
config=config)
infer.infer2file()
config.logger.info("Finished Test.")
