Bartsmiles

🚀 BARTSmiles 代碼庫

本代碼庫基於 https://huggingface.co/chenxran/bart-smiles/tree/main 進行適配，提供了使用 BARTSmiles 模型進行分子特徵提取的示例代碼。

🚀 快速開始

以下是使用 BARTSmiles 模型進行分子特徵提取的示例代碼：

基礎用法

from transformers import AutoTokenizer, AutoModel, SequenceFeatureExtractor
import torch
from transformers import AutoTokenizer, AutoModel

smiles = "CCC(=O)"

tokenizer = AutoTokenizer.from_pretrained("./BARTSmiles/", add_prefix_space=True)
inputs = tokenizer(smiles, return_tensors="pt", return_token_type_ids=False, add_special_tokens=True)

model = AutoModel.from_pretrained('./BARTSmiles')
model.eval()

# Use a pipeline as a high-level helper
from transformers import pipeline

extractor = pipeline("feature-extraction", model=model, tokenizer=tokenizer)
result = extractor(smiles, return_tensors=True, tokenize_kwargs={'return_token_type_ids':False})

📄 引用

如果您使用了本代碼庫，請引用以下論文：

@article{chilingaryan2022bartsmiles,
  title={Bartsmiles: Generative masked language models for molecular representations},
  author={Chilingaryan, Gayane and Tamoyan, Hovhannes and Tevosyan, Ani and Babayan, Nelly and Khondkaryan, Lusine and Hambardzumyan, Karen and Navoyan, Zaven and Khachatrian, Hrant and Aghajanyan, Armen},
  journal={arXiv preprint arXiv:2211.16349},
  year={2022}
}