Skip to content

Cryptodataset

Bases: Dataset

Dataset class for the LSTM model used by PyTorch Lightning.

Source code in make_us_rich/pipelines/training/crypto_dataset.py
12
13
def __init__(self, sequences: List[Tuple[pd.DataFrame, float]]):
    self.sequences = sequences

Last update: 2022-05-04