load_trained_model
and instantiate_model_for_training
)particle_requirement
, query_particle_requirement
and target_requirement
utils/graphutils
TripletGNNBase
, the base class of all triplet-based GNNsutils/graphutils
EdgeBasedGNN
and SciFiTripletInteractionGNN
triplet-based GNNs