pub struct FewShotLearner { /* private fields */ }Expand description
Few-shot learner for quick adaptation.
Given a small number of examples, adapts quickly using meta-learned priors.
Implementations§
Source§impl FewShotLearner
impl FewShotLearner
Sourcepub fn from_meta(meta: &MetaLearner, task_embedding: Option<&[f32]>) -> Self
pub fn from_meta(meta: &MetaLearner, task_embedding: Option<&[f32]>) -> Self
Create from meta-learner initialization.
Sourcepub fn with_adapt_lr(self, lr: f32) -> Self
pub fn with_adapt_lr(self, lr: f32) -> Self
Set adaptation learning rate.
Sourcepub fn with_adapt_steps(self, steps: usize) -> Self
pub fn with_adapt_steps(self, steps: usize) -> Self
Set number of adaptation steps.
Sourcepub fn add_example(&mut self, features: Vec<f32>, target: f32)
pub fn add_example(&mut self, features: Vec<f32>, target: f32)
Add an example to the support set.
Sourcepub fn adapt(&mut self)
pub fn adapt(&mut self)
Adapt to the support set.
Performs gradient descent on support set to adapt parameters from meta-learned initialization.
Sourcepub fn get_adapted_params(&self) -> &[f32]
pub fn get_adapted_params(&self) -> &[f32]
Get final adapted parameters.
Sourcepub fn support_size(&self) -> usize
pub fn support_size(&self) -> usize
Number of support examples.
Trait Implementations§
Source§impl Clone for FewShotLearner
impl Clone for FewShotLearner
Source§fn clone(&self) -> FewShotLearner
fn clone(&self) -> FewShotLearner
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for FewShotLearner
impl Debug for FewShotLearner
Source§impl<'de> Deserialize<'de> for FewShotLearner
impl<'de> Deserialize<'de> for FewShotLearner
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for FewShotLearner
impl RefUnwindSafe for FewShotLearner
impl Send for FewShotLearner
impl Sync for FewShotLearner
impl Unpin for FewShotLearner
impl UnsafeUnpin for FewShotLearner
impl UnwindSafe for FewShotLearner
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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> IntoRequest<T> for T
impl<T> IntoRequest<T> for T
Source§fn into_request(self) -> Request<T>
fn into_request(self) -> Request<T>
Wrap the input message
T in a tonic::Request