Struct OfflineDatasetBuffer 

pub struct OfflineDatasetBuffer { /* private fields */ }

Read-only offline dataset buffer.

Implementations

impl OfflineDatasetBuffer

pub fn from_arrays( obs: Vec<f32>, next_obs: Vec<f32>, actions: Vec<f32>, rewards: Vec<f32>, terminated: Vec<u8>, truncated: Vec<u8>, obs_dim: usize, act_dim: usize, ) -> Result<Self, RloxError>

Create from flat arrays.

Arrays must be row-major: obs has length n * obs_dim, etc.

pub fn len(&self) -> usize

Number of transitions in the dataset.

pub fn is_empty(&self) -> bool

pub fn n_episodes(&self) -> usize

Number of episodes in the dataset.

pub fn obs_dim(&self) -> usize

pub fn act_dim(&self) -> usize

pub fn compute_normalization(&mut self)

Compute and cache normalization statistics.

pub fn sample(&self, batch_size: usize, seed: u64) -> OfflineBatch

Sample i.i.d. transitions uniformly.

pub fn sample_trajectories( &self, batch_size: usize, seq_len: usize, seed: u64, ) -> TrajectoryBatch

Sample contiguous trajectory subsequences.

Each sample is a contiguous window of seq_len transitions from a single episode. If the episode is shorter than seq_len, the sequence is right-padded with zeros and the mask indicates valid positions.

pub fn stats(&self) -> DatasetStats

Get dataset statistics.