Tahinli/optimization_algorithms
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: 🚑 first calculate all probabilities then send onlookers

Browse source
This commit is contained in:
Ahmet Kaan GÜMÜŞ 2024-11-20 00:20:40 +03:00
parent ed7407743f
commit 7525f92b0d
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
artificial_bee_colony/src/bee.rs
View file

@ -22,6 +22,11 @@ impl Bee {
.map(|food_source| food_source.fitness_calculation)
.sum::<f64>();
let probabilities = food_sources
.iter()
.map(|x| x.fitness_calculation / total_fitness)
.collect::<Vec<f64>>();
let onlooker_bee_count = input.food_source_number;
let mut where_to_look = 0;
for _ in 0..onlooker_bee_count {
@ -30,8 +35,7 @@ impl Bee {
where_to_look = 0;
}
let fitness_for_index = food_sources[where_to_look].fitness_calculation;
if rand::thread_rng().gen_range(0.0..=1.0) < fitness_for_index / total_fitness {
if rand::thread_rng().gen_range(0.0..=1.0) < probabilities[where_to_look] {
Bee::onlooker_bee(
food_sources,
where_to_look,