ray.data.DatasetPipeline.to_tf

DatasetPipeline.to_tf(*, output_signature: Union[tf.TypeSpec, List[tf.TypeSpec], Dict[str, tf.TypeSpec], Tuple[Union[tf.TypeSpec, List[tf.TypeSpec], Dict[str, tf.TypeSpec]], tf.TypeSpec]], label_column: Optional[str] = None, feature_columns: Optional[Union[List[str], List[List[str]], Dict[str, List[str]]]] = None, prefetch_blocks: int = 0, batch_size: int = 1, drop_last: bool = False) → tf.data.Dataset

Call Dataset.to_tf over the stream of output batches from the pipeline