pyepo.data.dataset

optDataset class based on PyTorch Dataset

Classes

optDataset(model, feats, costs)	This class is Torch Dataset for optimization problems.
optDatasetKNN(model, feats, costs[, k, weight])	This class is Torch Dataset for optimization problems, when using the robust kNN-loss.