scglue.models.glue.GLUETrainer.get_losses

GLUETrainer.get_losses(data, graph, data_batch_size=None, graph_batch_size=None, random_seed=None)[source]

Get loss values for given data

Parameters:

loader – Data loader

Return type:

typing.Mapping[str, float]

Returns:

loss_dict – Dict containing loss values