stock

Bases: time_series_dataloader

Source code in tinybig/data/time_series_dataloader.py

class stock(time_series_dataloader):
    def __init__(
        self,
        name: str = STOCK_DATA_PROFILE['name'],
        x_len: int = 10,
        y_len: int = 1,
        xy_gap: int = 1,
        coverage_year_range: int = 1,
        time_granularity: str = 'daily',
        target_attributes: str = 'Open',
        *args, **kwargs
    ):
        super().__init__(
            name=name,
            data_profile=STOCK_DATA_PROFILE,
            x_len=x_len, y_len=y_len, xy_gap=xy_gap,
            coverage_year_range=coverage_year_range,
            time_granularity=time_granularity,
            target_attributes=target_attributes,
            *args, **kwargs
        )