Struct AIBrain 

pub struct AIBrain {
    pub policy_net: Model,
    pub value_net: Model,
}

AI brain with policy and value networks.

Fields

policy_net: Model

value_net: Model

Implementations

impl AIBrain

pub fn new(state_dim: usize) -> Self

Create an AI brain for a given state dimension.

pub fn select_action(&self, state: &GameState, temperature: f32) -> Action

Select an action using softmax sampling with temperature. Higher temperature = more random; lower = more greedy.

pub fn evaluate_state(&self, state: &GameState) -> f32

Evaluate how favorable a state is (higher = better for AI).