pub struct Pinn { /* private fields */ }Expand description
A trained PINN for the harmonic oscillator u'' + ω² u = 0 on [0, t_max] with u(0)=1, u'(0)=0.
Implementations§
Source§impl Pinn
impl Pinn
Sourcepub fn new(
hidden: usize,
omega: f64,
t_max: f64,
n_colloc: usize,
seed: u64,
) -> Self
pub fn new( hidden: usize, omega: f64, t_max: f64, n_colloc: usize, seed: u64, ) -> Self
A PINN with the given hidden width, angular frequency omega, domain [0, t_max], and number of
interior collocation points.
Sourcepub fn exact(&self, t: f64) -> f64
pub fn exact(&self, t: f64) -> f64
The closed-form solution cos(ω t) (for verification / plotting the truth).
Sourcepub fn train_step(&mut self, lr: f64) -> f64
pub fn train_step(&mut self, lr: f64) -> f64
One Adam step; returns the loss before the step.
Auto Trait Implementations§
impl Freeze for Pinn
impl RefUnwindSafe for Pinn
impl Send for Pinn
impl Sync for Pinn
impl Unpin for Pinn
impl UnsafeUnpin for Pinn
impl UnwindSafe for Pinn
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
impl<T, U> Imply<T> for U
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 moreSource§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.