Struct border_tch_agent::iqn::IqnConfig

pub struct IqnConfig<F, M>where
    F: SubModel,
    M: SubModel,
    F::Config: DeserializeOwned + Serialize + Clone,
    M::Config: DeserializeOwned + Serialize + Clone + OutDim,{
    pub device: Option<Device>,
    /* private fields */
}

Configuration of Iqn agent.

Fields

device: Option<Device>

Implementations

impl<F, M> IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,

pub fn model_config( self, model_config: IqnModelConfig<F::Config, M::Config> ) -> Self

Sets the configuration of the model.

pub fn soft_update_interval(self, v: usize) -> Self

Set soft update interval.

pub fn n_updates_per_opt(self, v: usize) -> Self

Set numper of parameter update steps per optimization step.

pub fn min_transitions_warmup(self, v: usize) -> Self

Interval before starting optimization.

pub fn batch_size(self, v: usize) -> Self

Batch size.

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

Discount factor.

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

Soft update coefficient.

pub fn explorer(self, v: IqnExplorer) -> Self

Set explorer.

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

Sets the output dimention of the iqn model.

pub fn sample_percent_pred(self, v: IqnSample) -> Self

Sampling percent points.

pub fn sample_percent_tgt(self, v: IqnSample) -> Self

Sampling percent points.

pub fn sample_percent_act(self, v: IqnSample) -> Self

Sampling percent points.

pub fn device(self, device: Device) -> Self

Device.

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

Constructs IqnConfig from YAML file.

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

Saves IqnConfig.

Trait Implementations

impl<F, M> Clone for IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<F, M> Debug for IqnConfig<F, M>where F: SubModel + Debug, M: SubModel + Debug, F::Config: DeserializeOwned + Serialize + Clone + Debug, M::Config: DeserializeOwned + Serialize + Clone + OutDim + Debug,

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

Formats the value using the given formatter.

impl<F, M> Default for IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,

fn default() -> Self

Returns the “default value” for a type.

impl<'de, F, M> Deserialize<'de> for IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,

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

Deserialize this value from the given Serde deserializer.

impl<F, M> PartialEq<IqnConfig<F, M>> for IqnConfig<F, M>where F: SubModel + PartialEq, M: SubModel + PartialEq, F::Config: DeserializeOwned + Serialize + Clone + PartialEq, M::Config: DeserializeOwned + Serialize + Clone + OutDim + PartialEq,

fn eq(&self, other: &IqnConfig<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 IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,

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 IqnConfig<F, M>where F: SubModel, M: SubModel, F::Config: DeserializeOwned + Serialize + Clone, M::Config: DeserializeOwned + Serialize + Clone + OutDim,