Struct rust_bert::pipelines::sentence_embeddings::layers::DenseConfig

pub struct DenseConfig {
    pub in_features: i64,
    pub out_features: i64,
    pub bias: bool,
    pub activation_function: Activation,
}

Configuration for Dense layer.

Fields

in_features: i64

Size of the input dimension

out_features: i64

Output size

bias: bool

Add a bias vector

activation_function: Activation

Activation function applied on output

Trait Implementations

impl Config for DenseConfig

fn from_file<P: AsRef<Path>>(path: P) -> Self

Loads a Config object from a JSON file. The format is expected to be aligned with the Transformers library configuration files for each model. The parsing will fail if non-optional keys expected by the model are missing.

impl Debug for DenseConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DenseConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.