fix evaluation effiency mismatch between "graph_scoring_efficiency" and "gnn_efficiency_rz"
- Fix the mismatch between the efficiency value between
graph_scoring_efficiency
andgnn_efficiency_rz
, caused by the former actually plotting the cumulative efficiency and the latter the actual model efficiency. - Change the
gpus
argument in trainer ininfer_stage
todevices
for compatibility with latter versions of pytorch lightning. - Delete the
shuffle=True
setting in dataloader, because it messes up the number of train events when doing inference on multiple GPUs. - Add score distribution plot to the ROC curve plot (since these two plots have similar requirements)
- Some minor changes in plot labels.