Lizzy-Ok Oke posted an update — Sep 21, 2025 12:53 PM EDT # Usage: def get_classification_token(batch) -> torch.Tensor: """Process embeddings and return classification token.""" model = ProteinInteractionTransformer() return model(batch, num_embeddings=len(batch), embedding_dimensions=batch[0].size(0))
Lizzy-Ok Oke posted an update — Sep 21, 2025 12:53 PM EDT # Usage: def get_classification_token(batch) -> torch.Tensor: """Process embeddings and return classification token.""" model = ProteinInteractionTransformer() return model(batch, num_embeddings=len(batch), embedding_dimensions=batch[0].size(0))