Struct SGDOptimizer 

pub struct SGDOptimizer { /* private fields */ }

SGD optimizer with optional momentum and weight decay

Implementations

impl SGDOptimizer

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

impl SGDOptimizer

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

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

impl SGDOptimizer

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

Create a new SGD optimizer

Arguments

• param_count - Number of parameters
• learning_rate - Learning rate

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

Create with momentum

pub fn with_weight_decay( param_count: u32, learning_rate: f64, momentum: f64, weight_decay: f64, ) -> Self

Create with momentum and weight decay

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

Perform an optimization step

Arguments

• params - Parameter array
• gradients - Gradient array

Returns

Updated parameter array

pub fn reset(&mut self)

Reset optimizer state

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 SGDOptimizer

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

Safety

impl FromNapiRef for SGDOptimizer

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

Safety

impl FromNapiValue for &SGDOptimizer

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 SGDOptimizer

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 SGDOptimizer

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

impl ToNapiValue for SGDOptimizer

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

Safety

impl TypeName for &SGDOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut SGDOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for SGDOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &SGDOptimizer

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

Safety

impl ValidateNapiValue for &mut SGDOptimizer

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

Safety