class_docstr: Runs the training of a reward generation model which will be able to
  distinguish between graphs with different labels. Check
  :obj:`examples.auggraph.GraphAug.run_reward_gen` for examples on how
  to run the reward generation model.
  
  Args:
      data_root_path (string): Directory where datasets should be saved.
      dataset_name (:class:`dig.auggraph.method.GraphAug.constants.enums.DatasetName`):
          Name of the graph dataset.
      conf (dict): Hyperparameters for the model. Check
          :obj:`examples.auggraph.GraphAug.conf.reward_gen_conf` for examples
          on how to define the conf dictionary for the reward generator.
code: def test(self, loader):
      self.model.eval()
      num_correct, num_pos_correct, num_neg_correct = 0, 0, 0
  
      with torch.no_grad():
          for data_batch in loader:
              anchor_data, pos_data, neg_data = data_batch
              anchor_data, pos_data, neg_data = (
                  anchor_data.to(self.device),
                  pos_data.to(self.device),
                  neg_data.to(self.device),
              )
  
              output = self.model(anchor_data, pos_data)
              pred = (output.view(-1) > 0.5).long()
              num_correct += pred.sum().item()
              num_pos_correct += pred.sum().item()
  
              output = self.model(anchor_data, neg_data)
              pred = (output.view(-1) < 0.5).long()
              num_correct += pred.sum().item()
              num_neg_correct += pred.sum().item()
  
      return (
          num_correct / (2 * len(loader.dataset)),
          num_pos_correct / len(loader.dataset),
          num_neg_correct / len(loader.dataset),
      )
  
file_path: DIG/dig/auggraph/method/GraphAug/runner_reward_gen.py
imports: import os
  import torch
  import torch.nn.functional
  from torch_geometric.loader import DataLoader
  from torch_geometric.datasets import TUDataset
  from .model import RewardGenModel
  from dig.auggraph.dataset.aug_dataset import DegreeTrans, TripleSet
  from dig.auggraph.method.GraphAug.constants import *
label_desc: Use image node_id 5 for calling the RunnerRewardGen obj's underlying member method code with example usage: obj.test(loader) and returns: unknown,  unknown,  unknown