feat(selection): add random and single-objective tournament selection
`select_random` samples `count` decisions with replacement and clones them out of the population (spec §10.1). `tournament_select_single_objective` runs binary-or-larger tournaments with the spec's tiebreak order: feasible beats infeasible, lower violation among infeasibles, and direction-correct objective comparison among feasibles. Panics if not exactly one objective (spec §10.2). Selection helpers stay under `heuropt::selection` and are not part of the prelude (spec §15).
This commit is contained in:
@@ -6,4 +6,5 @@ pub mod core;
|
||||
pub mod operators;
|
||||
pub mod pareto;
|
||||
pub mod prelude;
|
||||
pub mod selection;
|
||||
pub mod traits;
|
||||
|
||||
@@ -0,0 +1,7 @@
|
||||
//! Reusable selection helpers used by built-in algorithms.
|
||||
|
||||
pub mod random;
|
||||
pub mod tournament;
|
||||
|
||||
pub use random::*;
|
||||
pub use tournament::*;
|
||||
@@ -0,0 +1,67 @@
|
||||
//! Uniform random selection (with replacement).
|
||||
|
||||
use rand::Rng as _;
|
||||
|
||||
use crate::core::candidate::Candidate;
|
||||
use crate::core::rng::Rng;
|
||||
|
||||
/// Select `count` decisions uniformly at random with replacement.
|
||||
///
|
||||
/// Returns cloned decisions. Panics if `population` is empty and `count > 0`
|
||||
/// (spec §10.1).
|
||||
pub fn select_random<D: Clone>(
|
||||
population: &[Candidate<D>],
|
||||
count: usize,
|
||||
rng: &mut Rng,
|
||||
) -> Vec<D> {
|
||||
if count == 0 {
|
||||
return Vec::new();
|
||||
}
|
||||
assert!(
|
||||
!population.is_empty(),
|
||||
"select_random called on empty population with count > 0",
|
||||
);
|
||||
let mut out = Vec::with_capacity(count);
|
||||
for _ in 0..count {
|
||||
let idx = rng.random_range(0..population.len());
|
||||
out.push(population[idx].decision.clone());
|
||||
}
|
||||
out
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::core::evaluation::Evaluation;
|
||||
use crate::core::rng::rng_from_seed;
|
||||
|
||||
fn cand(d: u32) -> Candidate<u32> {
|
||||
Candidate::new(d, Evaluation::new(vec![d as f64]))
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn returns_count_decisions_from_population() {
|
||||
let pop = [cand(1), cand(2), cand(3)];
|
||||
let mut rng = rng_from_seed(42);
|
||||
let picks = select_random(&pop, 5, &mut rng);
|
||||
assert_eq!(picks.len(), 5);
|
||||
for p in &picks {
|
||||
assert!([1, 2, 3].contains(p));
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn count_zero_returns_empty() {
|
||||
let pop = [cand(1)];
|
||||
let mut rng = rng_from_seed(0);
|
||||
assert!(select_random(&pop, 0, &mut rng).is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[should_panic(expected = "empty population")]
|
||||
fn empty_population_panics() {
|
||||
let pop: Vec<Candidate<u32>> = Vec::new();
|
||||
let mut rng = rng_from_seed(0);
|
||||
let _ = select_random(&pop, 1, &mut rng);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,130 @@
|
||||
//! Single-objective tournament selection.
|
||||
|
||||
use rand::Rng as _;
|
||||
|
||||
use crate::core::candidate::Candidate;
|
||||
use crate::core::objective::{Direction, ObjectiveSpace};
|
||||
use crate::core::rng::Rng;
|
||||
|
||||
/// Tournament selection for single-objective problems.
|
||||
///
|
||||
/// Each tournament samples `tournament_size` candidates uniformly with
|
||||
/// replacement; the best one's decision is cloned into the output. Tiebreak
|
||||
/// rules (spec §10.2):
|
||||
///
|
||||
/// 1. Feasible candidates beat infeasible candidates.
|
||||
/// 2. Among infeasibles, smaller `constraint_violation` wins.
|
||||
/// 3. Among feasibles, the direction-correct best objective wins.
|
||||
///
|
||||
/// # Panics
|
||||
/// If `objectives` does not contain exactly one objective, or if `population`
|
||||
/// is empty when `count > 0`, or if `tournament_size == 0`.
|
||||
pub fn tournament_select_single_objective<D: Clone>(
|
||||
population: &[Candidate<D>],
|
||||
objectives: &ObjectiveSpace,
|
||||
tournament_size: usize,
|
||||
count: usize,
|
||||
rng: &mut Rng,
|
||||
) -> Vec<D> {
|
||||
assert!(
|
||||
objectives.is_single_objective(),
|
||||
"tournament_select_single_objective requires exactly one objective",
|
||||
);
|
||||
assert!(
|
||||
tournament_size > 0,
|
||||
"tournament_size must be greater than 0",
|
||||
);
|
||||
if count == 0 {
|
||||
return Vec::new();
|
||||
}
|
||||
assert!(
|
||||
!population.is_empty(),
|
||||
"tournament_select_single_objective called on empty population with count > 0",
|
||||
);
|
||||
|
||||
let direction = objectives.objectives[0].direction;
|
||||
let mut out = Vec::with_capacity(count);
|
||||
for _ in 0..count {
|
||||
let mut best_idx = rng.random_range(0..population.len());
|
||||
for _ in 1..tournament_size {
|
||||
let challenger = rng.random_range(0..population.len());
|
||||
if challenger_wins(&population[challenger], &population[best_idx], direction) {
|
||||
best_idx = challenger;
|
||||
}
|
||||
}
|
||||
out.push(population[best_idx].decision.clone());
|
||||
}
|
||||
out
|
||||
}
|
||||
|
||||
fn challenger_wins<D>(c: &Candidate<D>, b: &Candidate<D>, dir: Direction) -> bool {
|
||||
match (c.evaluation.is_feasible(), b.evaluation.is_feasible()) {
|
||||
(true, false) => true,
|
||||
(false, true) => false,
|
||||
(false, false) => c.evaluation.constraint_violation < b.evaluation.constraint_violation,
|
||||
(true, true) => {
|
||||
let cv = c.evaluation.objectives.first().copied().unwrap_or(f64::INFINITY);
|
||||
let bv = b.evaluation.objectives.first().copied().unwrap_or(f64::INFINITY);
|
||||
match dir {
|
||||
Direction::Minimize => cv < bv,
|
||||
Direction::Maximize => cv > bv,
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::core::evaluation::Evaluation;
|
||||
use crate::core::objective::Objective;
|
||||
use crate::core::rng::rng_from_seed;
|
||||
|
||||
fn cand_min(d: u32, v: f64) -> Candidate<u32> {
|
||||
Candidate::new(d, Evaluation::new(vec![v]))
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn large_tournament_picks_best_minimize() {
|
||||
let s = ObjectiveSpace::new(vec![Objective::minimize("f")]);
|
||||
let pop = [cand_min(1, 10.0), cand_min(2, 1.0), cand_min(3, 5.0)];
|
||||
let mut rng = rng_from_seed(1);
|
||||
// Tournament size equal to population almost always returns the best.
|
||||
let picks = tournament_select_single_objective(&pop, &s, 100, 10, &mut rng);
|
||||
assert!(picks.iter().all(|&d| d == 2));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn large_tournament_picks_best_maximize() {
|
||||
let s = ObjectiveSpace::new(vec![Objective::maximize("score")]);
|
||||
let pop = [cand_min(1, 10.0), cand_min(2, 1.0), cand_min(3, 5.0)];
|
||||
let mut rng = rng_from_seed(2);
|
||||
let picks = tournament_select_single_objective(&pop, &s, 100, 10, &mut rng);
|
||||
assert!(picks.iter().all(|&d| d == 1));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn feasible_beats_infeasible() {
|
||||
let s = ObjectiveSpace::new(vec![Objective::minimize("f")]);
|
||||
let pop = [
|
||||
Candidate::new(1u32, Evaluation::constrained(vec![0.0], 5.0)),
|
||||
Candidate::new(2u32, Evaluation::new(vec![100.0])),
|
||||
];
|
||||
let mut rng = rng_from_seed(3);
|
||||
let picks = tournament_select_single_objective(&pop, &s, 50, 20, &mut rng);
|
||||
// Feasible candidate (decision 2) wins regardless of objective value.
|
||||
assert!(picks.iter().all(|&d| d == 2));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[should_panic(expected = "exactly one objective")]
|
||||
fn multi_objective_panics() {
|
||||
let s = ObjectiveSpace::new(vec![
|
||||
Objective::minimize("f1"),
|
||||
Objective::minimize("f2"),
|
||||
]);
|
||||
let pop = [cand_min(1, 1.0)];
|
||||
let mut rng = rng_from_seed(0);
|
||||
let _ = tournament_select_single_objective(&pop, &s, 2, 1, &mut rng);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user