ml3/mbrl_utils.py [16:26]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
class Dataset(torch.utils.data.Dataset):
    def __init__(self, x, y):
        self.dataset = [
            (torch.FloatTensor(x[i]), torch.FloatTensor(y[i])) for i in range(len(x))
        ]

    def __len__(self):
        return len(self.dataset)

    def __getitem__(self, idx):
        return self.dataset[idx]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



ml3/mbrl_utils.py [110:120]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
class Dataset(torch.utils.data.Dataset):
    def __init__(self, x, y):
        self.dataset = [
            (torch.FloatTensor(x[i]), torch.FloatTensor(y[i])) for i in range(len(x))
        ]

    def __len__(self):
        return len(self.dataset)

    def __getitem__(self, idx):
        return self.dataset[idx]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -