Extract Embeddings

#1

Hello,
I have a question about how to extract embedding from the last layer of GNN model (GAT for example)?
Someone know how to do that or if there is a function to do this.
Thank you

#2

Hi @Ons_Masmoudi,

The GAT model in DeepChem doesn’t include an explicit method to obtain embeddings from the last layer, as it relies on the GATPredictor implementation in DGL Life Sciences (dglife link). I recommend creating a custom wrapper around GATPredictor to return graph_feats for this purpose.