Struct TransformerModel

pub struct TransformerModel<T = f64> {
    pub config: StandardModelConfig<T>,
    pub features: ModelFeatures,
    pub params: DeepModelParams<T>,
}

Fields

config: StandardModelConfig<T>

features: ModelFeatures

params: DeepModelParams<T>

Implementations

impl<T> TransformerModel<T>

pub fn new(config: StandardModelConfig<T>, features: ModelFeatures) -> Self

where T: Clone + Default,

pub fn init(self) -> Self

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

pub const fn config(&self) -> &StandardModelConfig<T>

returns a reference to the model configuration

pub const fn config_mut(&mut self) -> &mut StandardModelConfig<T>

returns a mutable reference to the model configuration

pub const fn features(&self) -> ModelFeatures

returns the model features

pub const fn features_mut(&mut self) -> &mut ModelFeatures

returns a mutable reference to the model features

pub const fn params(&self) -> &DeepModelParams<T>

returns a reference to the model parameters

pub const fn params_mut(&mut self) -> &mut DeepModelParams<T>

returns a mutable reference to the model parameters

pub fn set_config(&mut self, config: StandardModelConfig<T>) -> &mut Self

set the current configuration and return a mutable reference to the model

pub fn set_features(&mut self, features: ModelFeatures) -> &mut Self

set the current features and return a mutable reference to the model

pub fn set_params(&mut self, params: DeepModelParams<T>) -> &mut Self

set the current parameters and return a mutable reference to the model

pub fn with_config(self, config: StandardModelConfig<T>) -> Self

consumes the current instance to create another with the given configuration

pub fn with_features(self, features: ModelFeatures) -> Self

consumes the current instance to create another with the given features

pub fn with_params(self, params: DeepModelParams<T>) -> Self

consumes the current instance to create another with the given parameters

Trait Implementations

impl<T: Clone> Clone for TransformerModel<T>

fn clone(&self) -> TransformerModel<T>

Returns a duplicate of the value.

const fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug> Debug for TransformerModel<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<A, U, V> Forward<U> for TransformerModel<A>

where A: Float + FromPrimitive + ScalarOperand, V: ReLU<Output = V> + Sigmoid<Output = V> + for<'a> Add<&'a Array1<A>, Output = V>, Params<A>: Forward<U, Output = V> + Forward<V, Output = V>, for<'a> &'a U: Dot<Array2<A>, Output = V> + Add<&'a Array1<A>>,

type Output = V

fn forward(&self, input: &U) -> Result<Self::Output>

a single forward step

fn forward_then<F>(&self, input: &Rhs, then: F) -> Result<Self::Output, Error>

where F: FnOnce(Self::Output) -> Self::Output,

this method enables the forward pass to be generically activated using some closure. This is useful for isolating the logic of the forward pass from that of the activation function and is often used by layers and models.

impl<T> Model<T> for TransformerModel<T>

type Config = StandardModelConfig<T>

The type of configuration used for the model

type Layout = ModelFeatures

The type of ModelLayout used by the model for this implementation.

fn config(&self) -> &StandardModelConfig<T>

returns an immutable reference to the models configuration; this is typically used to access the models hyperparameters (i.e. learning rate, momentum, etc.) and other related control parameters.

fn config_mut(&mut self) -> &mut StandardModelConfig<T>

returns a mutable reference to the models configuration; useful for setting hyperparams

fn layout(&self) -> ModelFeatures

returns a copy of the model’s current layout (features); a type providing the model with a particular number of features for the various layers of a deep neural network.

fn params(&self) -> &DeepModelParams<T>

returns an immutable reference to the model parameters

fn params_mut(&mut self) -> &mut DeepModelParams<T>

returns a mutable reference to the model’s parameters

fn predict<U, V>(&self, inputs: &U) -> Result<V, ModelError>

where Self: Predict<U, Output = V>,

propagates the input through the model; each layer is applied in sequence meaning that the output of each previous layer is the input to the next layer. This pattern repeats until the output layer returns the final result.

fn train<U, V, W>( &mut self, dataset: &DatasetBase<U, V>, ) -> Result<W, ModelError>

where Self: Train<U, V, Output = W>,

a convience method that trains the model using the provided dataset; this method requires that the model implements the Train trait and that the dataset

impl<A, S, T> Train<ArrayBase<S, Dim<[usize; 1]>>, ArrayBase<T, Dim<[usize; 1]>>> for TransformerModel<A>

where A: Float + FromPrimitive + NumAssign + ScalarOperand + Debug, S: Data<Elem = A>, T: Data<Elem = A>,

type Output = A

fn train( &mut self, input: &ArrayBase<S, Ix1>, target: &ArrayBase<T, Ix1>, ) -> Result<Self::Output, ModelError>

fn train_for( &mut self, input: &X, target: &Y, epochs: usize, ) -> Result<Self::Output, ModelError>

impl<A, S, T> Train<ArrayBase<S, Dim<[usize; 2]>>, ArrayBase<T, Dim<[usize; 2]>>> for TransformerModel<A>

where A: Float + FromPrimitive + NumAssign + ScalarOperand + Debug, S: Data<Elem = A>, T: Data<Elem = A>,

type Output = A

fn train( &mut self, input: &ArrayBase<S, Ix2>, target: &ArrayBase<T, Ix2>, ) -> Result<Self::Output, ModelError>

fn train_for( &mut self, input: &X, target: &Y, epochs: usize, ) -> Result<Self::Output, ModelError>