gluonts.dataset.arrow.enc module#
- class gluonts.dataset.arrow.enc.ArrowEncoder(columns: List[str], ndarray_columns: Set[str] = <factory>, flatten_arrays: bool = True)[source]#
Bases:
object
- columns: List[str]#
- flatten_arrays: bool = True#
- ndarray_columns: Set[str]#
- class gluonts.dataset.arrow.enc.ArrowWriter(stream: bool = False, suffix: str = '.feather', compression: Union[typing_extensions.Literal['lz4'], typing_extensions.Literal['zstd'], NoneType] = None, flatten_arrays: bool = True, metadata: Union[dict, NoneType] = None)[source]#
Bases:
gluonts.dataset.DatasetWriter
- compression: Optional[Union[typing_extensions.Literal[lz4], typing_extensions.Literal[zstd]]] = None#
- flatten_arrays: bool = True#
- metadata: Optional[dict] = None#
- stream: bool = False#
- suffix: str = '.feather'#
- write_to_file(dataset: gluonts.dataset.Dataset, path: pathlib.Path) None [source]#
- write_to_folder(dataset: gluonts.dataset.Dataset, folder: pathlib.Path, name: Optional[str] = None) None [source]#
- class gluonts.dataset.arrow.enc.ParquetWriter(suffix: str = '.parquet', flatten_arrays: bool = True, metadata: Union[dict, NoneType] = None)[source]#
Bases:
gluonts.dataset.DatasetWriter
- flatten_arrays: bool = True#
- metadata: Optional[dict] = None#
- suffix: str = '.parquet'#
- write_to_file(dataset: gluonts.dataset.Dataset, path: pathlib.Path) None [source]#
- write_to_folder(dataset: gluonts.dataset.Dataset, folder: pathlib.Path, name: Optional[str] = None) None [source]#