ray.data.Dataset.random_sample#

Dataset.random_sample(fraction: float, *, seed: Optional[int] = None) → ray.data.dataset.Dataset[ray.data.block.T][source]#

Randomly samples a fraction of the elements of this dataset.

Note that the exact number of elements returned is not guaranteed, and that the number of elements being returned is roughly fraction * total_rows.

Examples

>>> import ray
>>> ds = ray.data.range(100) 
>>> ds.random_sample(0.1) 
>>> ds.random_sample(0.2, seed=12345)
Parameters

  • fraction – The fraction of elements to sample.

  • seed – Seeds the python random pRNG generator.

Returns

Returns a Dataset containing the sampled elements.