Struct border_tch_agent::iqn::IqnModelConfig

pub struct IqnModelConfig<F, M> where
    M: OutDim,  {
    pub feature_dim: i64,
    pub embed_dim: i64,
    pub f_config: Option<F>,
    pub m_config: Option<M>,
    pub opt_config: OptimizerConfig,
}

Configuration of IqnModel.

The type parameter F represents a configuration struct of a feature extractor. The type parameter M represents a configuration struct of a model for merging cosine-embedded percent points and feature vectors.

Fields

feature_dim: i64

Dimension of feature vector.

embed_dim: i64

Embedding dimension.

f_config: Option<F>

Configuration of feature extractor.

m_config: Option<M>

Configuration of a model for merging percentils and feature vectors.

opt_config: OptimizerConfig

Configuration of optimizer.

Implementations

impl<F, M> IqnModelConfig<F, M> where
    F: DeserializeOwned + Serialize,
    M: DeserializeOwned + Serialize + OutDim, 

pub fn learning_rate(self, v: f64) -> Self

Sets the learning rate.

impl<F, M> IqnModelConfig<F, M> where
    F: DeserializeOwned + Serialize,
    M: DeserializeOwned + Serialize + OutDim, 

pub fn embed_dim(self, v: i64) -> Self

Sets the dimension of cos-embedding of percent points.

pub fn feature_dim(self, v: i64) -> Self

Sets the dimension of feature vectors.

pub fn f_config(self, v: F) -> Self

Sets configurations for feature extractor.

pub fn m_config(self, v: M) -> Self

Sets configurations for output model.

pub fn out_dim(self, out_dim: i64) -> Self

Sets output dimension of the model.

pub fn opt_config(self, v: OptimizerConfig) -> Self

Sets optimizer configuration.

pub fn load(path: impl AsRef<Path>) -> Result<Self>

Constructs IqnModelConfig from YAML file.

pub fn save(&self, path: impl AsRef<Path>) -> Result<()>

Saves IqnModelConfig.

Trait Implementations

impl<F: Clone, M: Clone> Clone for IqnModelConfig<F, M> where
    M: OutDim, 

fn clone(&self) -> IqnModelConfig<F, M>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<F: Debug, M: Debug> Debug for IqnModelConfig<F, M> where
    M: OutDim, 

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

Formats the value using the given formatter.

impl<F, M> Default for IqnModelConfig<F, M> where
    M: OutDim, 

fn default() -> Self

Returns the “default value” for a type.

impl<'de, F, M> Deserialize<'de> for IqnModelConfig<F, M> where
    M: OutDim,
    F: Deserialize<'de>,
    M: Deserialize<'de>, 

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.

impl<F: PartialEq, M: PartialEq> PartialEq<IqnModelConfig<F, M>> for IqnModelConfig<F, M> where
    M: OutDim, 

fn eq(&self, other: &IqnModelConfig<F, M>) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &IqnModelConfig<F, M>) -> bool

This method tests for !=.

impl<F, M> Serialize for IqnModelConfig<F, M> where
    M: OutDim,
    F: Serialize,
    M: Serialize, 

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where
    __S: Serializer, 

Serialize this value into the given Serde serializer.

impl<F, M> StructuralPartialEq for IqnModelConfig<F, M> where
    M: OutDim,