请问如何用该模型预测我的数据？

[wangmatlab]

No description provided.

[wangmatlab]

训练好了这个模型 如何用来预测我的数据？

[lemonhu]

谢谢关注。
建议参考文件evaluate.py，即可实现模型的测试评估。

[wangmatlab]

谢谢关注。
建议参考文件evaluate.py，即可实现模型的测试评估。
感谢作者回信，
我是想用该模型预测任意输入的文本（而不是用于测试已有的测试数据），请问该如何实现？

[lemonhu]

利用训练好的模型提供离线/在线服务，结合data_loader.py和evaluate.py，进行适当改写即可了。

[temco]

类似于这样，在evaluate.py中添加一个函数

def predict(line, model, data_loader, params): # added by temco for prediction
    """predict the sentence with the model"""
    idx2tag = params.idx2tag
    pred_tags = []
    batch_data = []
    tokens = data_loader.tokenizer.tokenize(line.strip())
    batch_data.append(data_loader.tokenizer.convert_tokens_to_ids(tokens))
    batch_data = torch.tensor(batch_data, dtype=torch.long)
    batch_masks = batch_data.gt(0)
    batch_output = model(batch_data, token_type_ids=None, attention_mask=batch_masks)
    batch_output = batch_output.detach().cpu().numpy()
    pred_tags = []
    pred_tags.extend([idx2tag.get(idx) for indices in np.argmax(batch_output, axis=2) for idx in indices])
    logging.info(pred_tags)