DeepChainBio / bio-transformers

bio-transformers is a wrapper on top of the ESM/Protbert model, trained on millions on proteins and used to predict embeddings.
https://bio-transformers.readthedocs.io/en/latest/getting_started/install.html
Apache License 2.0
143 stars 31 forks source link

add options to save the embedding #5

Open delfosseaurelien opened 3 years ago

delfosseaurelien commented 3 years ago

save the embeddings with the format :

f"{variable_name}_{model_name}_{embeddings_type}_embeddings.npy"