Type Alias ShallowParamsBase

pub type ShallowParamsBase<S, D> = ModelParamsBase<S, D, ParamsBase<S, D>>;

a type alias for a shallow representation of the ModelParamsBase using a single ParamsBase instance as the hidden layer.

Aliased Type

pub struct ShallowParamsBase<S, D> { /* private fields */ }

Implementations

impl<S, D, A> ShallowParamsBase<S, D>

where S: RawData<Elem = A>, D: Dimension,

pub fn default(input: D, hidden: D, output: D) -> Self

where A: Clone + Default, S: DataOwned, D: RemoveAxis,

initialize a new instance of the ShallowParamsBase with the given input, hidden, and output dimensions using the default values for the parameters

pub fn size(&self) -> usize

returns the total number parameters within the model, including the input and output layers

pub const fn hidden_weights(&self) -> &ArrayBase<S, D>

returns an immutable reference to the hidden weights

pub const fn hidden_weights_mut(&mut self) -> &mut ArrayBase<S, D>

returns an mutable reference to the hidden weights

impl<S, A> ShallowParamsBase<S, Ix2>

where S: RawData<Elem = A>,

pub fn from_features(features: ModelFeatures) -> Self

where A: Clone + Default, S: DataOwned,

pub fn forward(&self, input: &Array1<A>) -> Result<Array1<A>>

where A: Float + ScalarOperand, S: Data,

forward input through the controller network

impl<A, S> ShallowParamsBase<S, Ix2>

where S: DataOwned<Elem = A>,

pub fn init(self) -> Self

where A: Float + FromPrimitive + SampleUniform, StandardNormal: Distribution<A>,

consumes the controller to initialize the various parameters with random values

pub fn init_rand<G, Ds>(features: ModelFeatures, distr: G) -> Self

where G: Fn((usize, usize)) -> Ds, Ds: Clone + Distribution<A>, S: DataOwned,

returns a new instance of the model initialized with the given features and random distribution

pub fn glorot_normal(features: ModelFeatures) -> Self

where A: Float + FromPrimitive, StandardNormal: Distribution<A>,

initialize the model parameters using a glorot normal distribution

pub fn glorot_uniform(features: ModelFeatures) -> Self

where A: Float + FromPrimitive + SampleUniform, <A as SampleUniform>::Sampler: Clone, Uniform<A>: Distribution<A>,

initialize the model parameters using a glorot uniform distribution

Trait Implementations

impl<A, S> Default for ShallowParamsBase<S, Ix2>

where S: DataOwned<Elem = A>, A: Clone + Default,

fn default() -> Self

Returns the “default value” for a type.