pub struct PrioritizedReplayBuffer { /* private fields */ }Expand description
Prioritized replay buffer
Implementations§
Source§impl PrioritizedReplayBuffer
impl PrioritizedReplayBuffer
Sourcepub fn new(capacity: usize, alpha: f64, beta: f64) -> Self
pub fn new(capacity: usize, alpha: f64, beta: f64) -> Self
Create new prioritized replay buffer
Sourcepub fn add(&mut self, experience: Experience)
pub fn add(&mut self, experience: Experience)
Add experience to buffer
Sourcepub fn sample(&self, batch_size: usize) -> Option<SampledBatch>
pub fn sample(&self, batch_size: usize) -> Option<SampledBatch>
Sample batch from buffer
Sourcepub fn update_priorities(&mut self, indices: &[usize], td_errors: &[f32])
pub fn update_priorities(&mut self, indices: &[usize], td_errors: &[f32])
Update priorities based on TD errors
Auto Trait Implementations§
impl Freeze for PrioritizedReplayBuffer
impl RefUnwindSafe for PrioritizedReplayBuffer
impl Send for PrioritizedReplayBuffer
impl Sync for PrioritizedReplayBuffer
impl Unpin for PrioritizedReplayBuffer
impl UnsafeUnpin for PrioritizedReplayBuffer
impl UnwindSafe for PrioritizedReplayBuffer
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more