pytupli.dataset

Module for everything related to dataset management.

Classes

BaseTupleParser

Base class for parsing (nested) lists of obs/act/reward/term/trunc into tensors for other formats like numpy arrays, PyTorch tensors, etc.

NumpyTupleParser

Parser for converting lists of observations, actions, rewards, terminal flags, and timeout flags into numpy arrays.

TensorflowTupleParser

Parser for converting lists of observations, actions, rewards, terminal flags, and timeout flags into TensorFlow tensors.

TorchTupleParser

Parser for converting lists of observations, actions, rewards, terminal flags, and timeout flags into PyTorch tensors.

TupliDataset

A dataset class for downloading, managing and filtering offline RL tuple data.