Datasets

class optexp.datasets.Dataset[source]

Abstract base class for datasets.

Vision datasets

class optexp.datasets.MNIST[source]

The MNIST dataset, provided through TorchVision.

Misc

class optexp.datasets.DummyRegression[source]

A dummy dataset for testing purposes.

The results is generated by a sine function:

y = x + 0.1 * sin(10 * 2 * pi * x)

The inputs x form a linear grid from 0 to 1 (linspace(0, 1, n_samples)).

Parameters:
  • n_tr (int, optional) – number of training samples. Defaults to 100.

  • n_va (int, optional) – number of validation samples. Defaults to 10.