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, require_solpool=False)

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

solve_ratio = 1

solpool = None

_solset

reduction = 'mean'

abstract forward(pred_cost, true_cost): Forward pass

_reduce(loss): Apply reduction to loss tensor