Roberta Large Conceptnet Mask Prompt E Nce

🚀 relbert/roberta-large-conceptnet-mask-prompt-e-nce

RelBERT 是在 roberta-large 基础上微调得到的模型，微调数据集为 relbert/conceptnet_high_confidence。微调过程借助 RelBERT 库完成（更多详情请查看该仓库）。该模型在关系理解任务中取得了以下成果：

🚀 快速开始

本模型可通过 relbert 库 使用。首先，使用以下命令安装该库：

pip install relbert

然后，按如下方式激活模型：

from relbert import RelBERT
model = RelBERT("relbert/roberta-large-conceptnet-mask-prompt-e-nce")
vector = model.get_embedding(['Tokyo', 'Japan'])  # shape of (1024, )

✨ 主要特性

该模型在多种关系理解任务中表现出色，具体如下：

📦 安装指南

通过 pip 安装 relbert 库：

pip install relbert

💻 使用示例

基础用法

from relbert import RelBERT
model = RelBERT("relbert/roberta-large-conceptnet-mask-prompt-e-nce")
vector = model.get_embedding(['Tokyo', 'Japan'])  # shape of (1024, )

📚 详细文档

训练超参数

训练过程中使用了以下超参数：

完整配置可在 微调参数文件 中查看。

参考引用

如果您使用了 RelBERT 的任何资源，请考虑引用我们的 论文。

@inproceedings{ushio-etal-2021-distilling-relation-embeddings,
    title = "{D}istilling {R}elation {E}mbeddings from {P}re-trained {L}anguage {M}odels",
    author = "Ushio, Asahi  and
      Schockaert, Steven  and
      Camacho-Collados, Jose",
    booktitle = "EMNLP 2021",
    year = "2021",
    address = "Online",
    publisher = "Association for Computational Linguistics",
}