paddlets.models.classify.dl.adapter.data_adapter

class ClassifyDataAdapter[源代码]

基类：object

数据转换类，用于将TSDataset数据集转换为paddle Dataset样本数据集和paddle DataLoader样本数据批量加载器。

to_paddle_dataset(rawdatasets: List[TSDataset], labels: ndarray) → ClassifyPaddleDatasetImpl[源代码]

将TSDataset数据集转换为paddle Dataset样本数据集。

参数

  • rawdataset (TSDataset) – 待转换的原始TSDataset数据集。

  • labels – 数据分类标签

返回

一个构建完成的用于分类模型的paddle Dataset样本数据集。

返回类型

PaddleDatasetImpl

to_paddle_dataloader(paddle_dataset: ClassifyPaddleDatasetImpl, batch_size: int, collate_fn: Optional[Callable] = None, shuffle: bool = True) → DataLoader[源代码]

将paddle Dataset样本数据集转换为DataLoader样本批量加载器。

参数

  • paddle_dataset (PaddleDatasetImpl) – 待转换的paddle Dataset样本数据集。

  • batch_size (int) – 单个数据批量中包含的样本条数。

  • collate_fn (Callable, optional) – 用户自定义的单批量样本整合函数，可选参数。

  • shuffle (bool, optional) – 是否在生成批量样本之前对数据进行洗牌。默认为True。

返回

一个构建完成的paddle样本批量数据加载器。

返回类型

PaddleDataLoader