def rollout_groundtruth()

in src/train.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

def rollout_groundtruth(problem, dataset, opts):
    return torch.cat([
        problem.get_costs(bat['nodes'], bat['tour_nodes'])[0]
        for bat in DataLoader(
            dataset, batch_size=opts.batch_size, shuffle=False, num_workers=opts.num_workers)
    ], 0)