pub struct Config {Show 37 fields
pub model_path: Option<PathBuf>,
pub site_profiles_dir: PathBuf,
pub output_dir: PathBuf,
pub models_dir: PathBuf,
pub use_cpu_for_tuning: bool,
pub algorithm: AlgorithmType,
pub num_episodes: usize,
pub batch_size: usize,
pub learning_rate: f64,
pub gamma: f64,
pub epsilon_start: f64,
pub epsilon_end: f64,
pub epsilon_decay: f64,
pub target_update_freq: usize,
pub max_steps_per_episode: usize,
pub replay_buffer_size: usize,
pub priority_alpha: f64,
pub priority_beta: f64,
pub min_replay_size: usize,
pub train_freq: usize,
pub num_train_steps_per_episode: usize,
pub max_html_samples: usize,
pub sample_batch_load_size: usize,
pub prefetch_samples: bool,
pub metrics_window: usize,
pub checkpoint_freq: usize,
pub log_freq: usize,
pub state_dim: usize,
pub num_discrete_actions: usize,
pub num_continuous_params: usize,
pub num_candidate_nodes: usize,
pub ppo_clip_epsilon: f32,
pub ppo_gae_lambda: f32,
pub ppo_value_loss_coef: f32,
pub ppo_entropy_coef: f32,
pub ppo_epochs: usize,
pub stopwords: HashSet<String>,
}Expand description
Configuration for the content extractor rl
Fields§
§model_path: Option<PathBuf>§site_profiles_dir: PathBuf§output_dir: PathBuf§models_dir: PathBuf§use_cpu_for_tuning: bool§algorithm: AlgorithmType§num_episodes: usize§batch_size: usize§learning_rate: f64§gamma: f64§epsilon_start: f64§epsilon_end: f64§epsilon_decay: f64§target_update_freq: usize§max_steps_per_episode: usize§replay_buffer_size: usize§priority_alpha: f64§priority_beta: f64§min_replay_size: usize§train_freq: usize§num_train_steps_per_episode: usize§max_html_samples: usize§sample_batch_load_size: usize§prefetch_samples: bool§metrics_window: usize§checkpoint_freq: usize§log_freq: usize§state_dim: usize§num_discrete_actions: usize§num_continuous_params: usize§num_candidate_nodes: usize§ppo_clip_epsilon: f32§ppo_gae_lambda: f32§ppo_value_loss_coef: f32§ppo_entropy_coef: f32§ppo_epochs: usize§stopwords: HashSet<String>Implementations§
Source§impl Config
impl Config
Sourcepub fn with_algorithm(algorithm: AlgorithmType) -> Self
pub fn with_algorithm(algorithm: AlgorithmType) -> Self
Create config with specific algorithm
Sourcepub fn ppo_recommended() -> Self
pub fn ppo_recommended() -> Self
PPO recommended configuration
Sourcepub fn dqn_optimized() -> Self
pub fn dqn_optimized() -> Self
DQN optimized configuration
Sourcepub fn gpu_optimized() -> Self
pub fn gpu_optimized() -> Self
Create high-performance GPU config
Sourcepub fn setup_directories(&self) -> Result<()>
pub fn setup_directories(&self) -> Result<()>
Setup required directories
Trait Implementations§
Source§impl<'de> Deserialize<'de> for Config
impl<'de> Deserialize<'de> for Config
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for Config
impl RefUnwindSafe for Config
impl Send for Config
impl Sync for Config
impl Unpin for Config
impl UnsafeUnpin for Config
impl UnwindSafe for Config
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more