File size: 417 Bytes
38af4fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
import pickle
# Example SMILES strings
smiles_list = ["CC(=O)OC1=CC=CC=C1C(=O)O", "C1CCCCC1", "CCN(CC)CC"]
# Create 256-D random embeddings
embeddings_dict = {smi: torch.randn(256) for smi in smiles_list}
# Save to embeddings.pkl
with open("data/processed/embeddings.pkl", "wb") as f:
pickle.dump(embeddings_dict, f)
print("Saved dummy embeddings to data/processed/embeddings.pkl")
|