Molt5 Large Smiles2caption

🚀 分子SMILES字符串轉描述模型

這個模型可用於根據SMILES字符串生成輸入的描述。

🚀 快速開始

此模型能夠依據SMILES字符串生成對應的描述。下面是使用該模型的詳細步驟和代碼示例。

💻 使用示例

基礎用法

from transformers import T5Tokenizer, T5ForConditionalGeneration

tokenizer = T5Tokenizer.from_pretrained("laituan245/molt5-large-smiles2caption", model_max_length=512)
model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-large-smiles2caption')

input_text = 'C1=CC2=C(C(=C1)[O-])NC(=CC2=O)C(=O)O'
input_ids = tokenizer(input_text, return_tensors="pt").input_ids

outputs = model.generate(input_ids, num_beams=5, max_length=512)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

📚 詳細文檔

論文參考

如需更多信息，請查看我們的論文。

• 論文：Translation between Molecules and Natural Language
• 作者：Carl Edwards*, Tuan Lai*, Kevin Ros, Garrett Honke, Heng Ji

📄 許可證

本項目採用Apache-2.0許可證。