ray/doc/source/ray-air/doc_code
Clark Zinzow a29baf93c8
[Datasets] Add .iter_torch_batches() and .iter_tf_batches() APIs. (#26689)
This PR adds .iter_torch_batches() and .iter_tf_batches() convenience APIs, which takes care of ML framework tensor conversion, the narrow tensor waste for the .iter_batches() call ("numpy" format), and unifies batch formats around two options: a single tensor for simple/pure-tensor/single-column datasets, and a dictionary of tensors for multi-column datasets.
2022-07-22 10:09:36 -07:00
..
air_ingest.py [air] Allow users to use instances of ScalingConfig (#25712) 2022-07-18 15:46:58 -07:00
air_key_concepts.py [air] Allow users to use instances of ScalingConfig (#25712) 2022-07-18 15:46:58 -07:00
checkpoint_mlflow.py [air/docs] checkpoints (#25901) 2022-07-11 20:40:23 -07:00
checkpoint_usage.py [AIR] Add framework-specific checkpoints (#26777) 2022-07-20 19:33:27 -07:00
config_scaling.py [air] Allow users to use instances of ScalingConfig (#25712) 2022-07-18 15:46:58 -07:00
output.txt [docs] Add AIR data ingest docs (part 1-- bulk loading only) (#24799) 2022-05-19 14:25:47 -07:00
predictors.py [Datasets] Add .iter_torch_batches() and .iter_tf_batches() APIs. (#26689) 2022-07-22 10:09:36 -07:00
preprocessors.py [air] Allow users to use instances of ScalingConfig (#25712) 2022-07-18 15:46:58 -07:00
pytorch_starter.py [air] Allow users to use instances of ScalingConfig (#25712) 2022-07-18 15:46:58 -07:00
tf_starter.py [Datasets] Add .iter_torch_batches() and .iter_tf_batches() APIs. (#26689) 2022-07-22 10:09:36 -07:00