aprender-core 0.31.2

Next-generation machine learning library in pure Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30


impl<S, P> SearchStrategy<P> for ActiveLearningSearch<S>
where
    S: SearchStrategy<P>,
    P: ParamKey,
{
    fn suggest(&mut self, space: &SearchSpace<P>, n: usize) -> Vec<Trial<P>> {
        // If we should stop, return empty
        if self.should_stop() {
            return Vec::new();
        }
        self.base.suggest(space, n)
    }

    fn update(&mut self, results: &[TrialResult<P>]) {
        // Collect scores for uncertainty estimation
        for result in results {
            self.scores.push(result.score);
        }

        // Update uncertainty estimate
        self.compute_uncertainty();

        // Forward to base strategy
        self.base.update(results);
    }
}

#[cfg(test)]
mod tests;