pyepo.func.abcmodule
Abstract autograd optimization module
Module Contents
Classes
An abstract module for the learning to rank losses, which measure the difference in how the predicted cost |
- class pyepo.func.abcmodule.optModule(optmodel, processes=1, solve_ratio=1, reduction='mean', dataset=None)
Bases:
torch.nn.ModuleAn 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.
- abstract forward(pred_cost, true_cost)
Forward pass
- _update_solution_pool(sol)
Add new solutions to solution pool