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 for IqnModelConfig<F, M>where M: OutDim + Clone,

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 for IqnModelConfig<F, M>where M: OutDim + Debug,

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 + Deserialize<'de>, F: 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<IqnModelConfig<F, M>> for IqnModelConfig<F, M>where M: OutDim + PartialEq,

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: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<F, M> Serialize for IqnModelConfig<F, M>where M: OutDim + Serialize, F: 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,