how to get protein embedding for Pfam, EC and GO models?

[hoangthienan95]

Thank you for a very cool model and paper. Reading the code, it's not immediately obvious how to get the embedding for each of the individual models. Also how to get the combined embedding for all three models? Apologize if I might have missed something obvious.

Thank you!

[zeyak]

@hoangthienan95 Have you already figured that out? I'd also like to perform an embedding analysis like the one provided on colab however, neither do I manage to run protein_dataset.py nor produce *.tfrecord outputs.