TUnA_embeddings / Intra2_dictionary /combine_dictionary.py
yk0's picture
Upload folder using huggingface_hub
1b9456c verified
raw
history blame contribute delete
402 Bytes
import torch
# Example dictionaries
dict1 = torch.load("/new-stg/home/young/ppi-data/gold_standard_embedded/Intra2_dictionary_1500_or_less/protein_dictionary.pt")
dict2 = torch.load("/new-stg/home/young/ppi-data/gold_standard_embedded/Intra2_dictionary_1500_greater/protein_dictionary.pt")
# Combine dictionaries
combined_dict = {**dict1, **dict2}
torch.save(combined_dict, "protein_dictionary.pt")