ICH-NOVA / data /processed /generate_dummy_embeddings.py
shreyab21's picture
Upload 6 files
38af4fb verified
raw
history blame contribute delete
417 Bytes
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")