Struct AdamOptimizer 

pub struct AdamOptimizer { /* private fields */ }

Adam optimizer with bias correction

Implementations

impl AdamOptimizer

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl AdamOptimizer

pub fn into_reference( val: AdamOptimizer, env: Env, ) -> Result<Reference<AdamOptimizer>>

pub fn into_instance(self, env: Env) -> Result<ClassInstance<AdamOptimizer>>

impl AdamOptimizer

pub fn new(param_count: u32, learning_rate: f64) -> Self

Create a new Adam optimizer

Arguments

• param_count - Number of parameters
• learning_rate - Learning rate

pub fn with_betas( param_count: u32, learning_rate: f64, beta1: f64, beta2: f64, ) -> Self

Create with custom betas

pub fn with_config( param_count: u32, learning_rate: f64, beta1: f64, beta2: f64, epsilon: f64, weight_decay: f64, ) -> Self

Create with full configuration

pub fn step( &mut self, params: Float32Array, gradients: Float32Array, ) -> Float32Array

Perform an optimization step

Returns

Updated parameter array

pub fn reset(&mut self)

Reset optimizer state (momentum terms)

pub fn learning_rate(&self) -> f64

Get current learning rate

pub fn set_learning_rate(&mut self, lr: f64)

Set learning rate

Trait Implementations

impl FromNapiMutRef for AdamOptimizer

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for AdamOptimizer

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &AdamOptimizer

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut AdamOptimizer

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for AdamOptimizer

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for AdamOptimizer

unsafe fn to_napi_value(env: napi_env, val: AdamOptimizer) -> Result<napi_value>

Safety

impl TypeName for &AdamOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut AdamOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for AdamOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &AdamOptimizer

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut AdamOptimizer

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety