现在将尝试提取医学概念之间的关系。根据经验,关系提取比命名实体提取至少难一个数量级。如果不希望命名实体链接取得完美的结果,那么关系提取技术肯定会出现一些错误。
使用zero-shot关系提取器FewRel。虽然我不建议将此模型投入生产,但它足以进行简单的演示。该模型可在HuggingFace上使用,因此我们不必进行培训或设置模型。
from transformers import AutoTokenizer
from zero_shot_re import RelTaggerModel, RelationExtractormodel = RelTaggerModel.from_pretrained("fractalego/fewrel-zero-shot")
tokenizer = AutoTokenizer.from_pretrained("bert-large-uncased-whole-word-masking-finetuned-squad")
relations = [