axtreme.dataΒΆ

Modules

batch_invariant_sampler

Contains samplers where the total dataset procuded is not effect by the size of the batch dimension used.

dataset

Contains objects and helpers for creating data unit testing.

fixed_random_sample

A copy of pytorch's RandomSampler that uses the same random sample for each iteration.

importance_dataset

Dataset that return importance sample information in the form (data, importance_weight).

multi_dim_batch_sampler

Allows retuning batches of data of an arbitrary dimension.

numpy_file_dataset

Datasets to help work with numpy files (.npy).

sizable_sequential_sample

Sequential sampling.