文件名称:specter:幽灵
文件大小:95KB
文件格式:ZIP
更新时间:2024-04-10 03:59:43
Python
演讲者:使用引用文献为基础的变形金刚进行文档级表示学习 ||| | || 该存储库包含代码,预训练模型的链接,使用以及评估框架的链接。 ***** 2021年1月新版:HuggingFace模型***** 现在可以通过HuggingFace的转换器库访问Spectre。 感谢提供了Huggingface培训脚本和检查点。 见下文: 如何使用预训练模型 1-通过Huggingface变形金刚库 要求: pip install --upgrade transformers==4.2 from transformers import AutoTokenizer , AutoModel # load model and tokenizer tokenizer = AutoTokenizer . from_pretrained ( 'allenai/specter' ) model = Au
【文件预览】:
specter-master
----scripts()
--------pytorch_lightning_training_script()
--------embed.py(2KB)
--------run-exp-simple.sh(4KB)
--------embed_papers_hf.py(3KB)
----requirements.txt(145B)
----LICENSE(11KB)
----specter()
--------data.py(24KB)
--------predict_command.py(9KB)
--------model.py(15KB)
--------__init__.py(139B)
--------predictor.py(4KB)
--------data_utils()
----setup.py(186B)
----README.md(9KB)
----data()
--------sample.ids(2KB)
--------sample-metadata.json(60KB)
--------training()
----.gitignore(2KB)
----experiment_configs()
--------simple.jsonnet(6KB)