pyepo.func.abcmodule

Abstract autograd optimization module

Classes

optModule

An abstract module for the learning to rank losses, which measure the difference in how the predicted cost

Module Contents

class pyepo.func.abcmodule.optModule(optmodel, processes=1, solve_ratio=1, reduction='mean', dataset=None)

Bases: torch.nn.Module

An abstract module for the learning to rank losses, which measure the difference in how the predicted cost vector and the true cost vector rank a pool of feasible solutions.

optmodel

processes = 1

solve_ratio = 1

solpool = None

reduction = 'mean'

abstract forward(pred_cost, true_cost): Forward pass

_update_solution_pool(sol): Add new solutions to solution pool