mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-25 13:30:12 +08:00
22 lines
677 B
Python
22 lines
677 B
Python
from typing import Iterable
|
|
|
|
from .common import DataEntry, Dataset, SourceContext
|
|
from .process import ProcessDataEntry
|
|
|
|
|
|
class ListDataset(Dataset):
|
|
def __init__(
|
|
self, data_iter: Iterable[DataEntry], freq: str, one_dim_target: bool = True
|
|
) -> None:
|
|
process = ProcessDataEntry(freq, one_dim_target)
|
|
self.list_data = [process(data) for data in data_iter]
|
|
|
|
def __iter__(self):
|
|
source_name = "list_data"
|
|
for row_number, data in enumerate(self.list_data, start=1):
|
|
data["source"] = SourceContext(source=source_name, row=row_number)
|
|
yield data
|
|
|
|
def __len__(self):
|
|
return len(self.list_data)
|