eduardosoares99 commited on
Commit
73acbc9
1 Parent(s): 4c21438

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -20,6 +20,8 @@ Welcome to IBM's series of large foundation models for sustainable materials. Ou
20
 
21
  GitHub: [GitHub Link](https://github.com/IBM/materials/tree/main)
22
 
 
 
23
  # SMILES-based Transformer Encoder-Decoder (SMI-TED)
24
 
25
  ![ted-smi](smi-ted.png)
@@ -167,4 +169,20 @@ For decoder, you can use the function, so you can return from embeddings to SMIL
167
  ```python
168
  with torch.no_grad():
169
  decoded_smiles = model.decode(encoded_embeddings)
170
- ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
  GitHub: [GitHub Link](https://github.com/IBM/materials/tree/main)
22
 
23
+ Paper: [arXiv:2407.20267](https://arxiv.org/abs/2407.20267)
24
+
25
  # SMILES-based Transformer Encoder-Decoder (SMI-TED)
26
 
27
  ![ted-smi](smi-ted.png)
 
169
  ```python
170
  with torch.no_grad():
171
  decoded_smiles = model.decode(encoded_embeddings)
172
+ ```
173
+
174
+
175
+ ## Citations
176
+
177
+ ```
178
+ @misc{soares2024largeencoderdecoderfamilyfoundation,
179
+ title={A Large Encoder-Decoder Family of Foundation Models For Chemical Language},
180
+ author={Eduardo Soares and Victor Shirasuna and Emilio Vital Brazil and Renato Cerqueira and Dmitry Zubarev and Kristin Schmidt},
181
+ year={2024},
182
+ eprint={2407.20267},
183
+ archivePrefix={arXiv},
184
+ primaryClass={cs.LG},
185
+ url={https://arxiv.org/abs/2407.20267},
186
+ }
187
+
188
+ ```