org.opt4j.tutorial.optimizer.MutateOptimizer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of opt4j-tutorial Show documentation

The tutorial classes for Opt4J

There is a newer version: 3.1.4

package org.opt4j.tutorial.optimizer;

import java.util.Collection;

import org.opt4j.core.Genotype;
import org.opt4j.core.Individual;
import org.opt4j.core.IndividualFactory;
import org.opt4j.core.optimizer.IterativeOptimizer;
import org.opt4j.core.optimizer.Population;
import org.opt4j.core.optimizer.TerminationException;
import org.opt4j.operators.copy.Copy;
import org.opt4j.operators.mutate.Mutate;
import org.opt4j.optimizers.ea.Selector;

import com.google.inject.Inject;

public class MutateOptimizer implements IterativeOptimizer {

	protected final IndividualFactory individualFactory;

	protected final Mutate mutate;

	protected final Copy copy;

	protected final Selector selector;

	private final Population population;

	public static final int POPSIZE = 100;

	public static final int OFFSIZE = 25;

	@Inject
	public MutateOptimizer(Population population, IndividualFactory individualFactory, Selector selector,
			Mutate mutate, Copy copy) {
		this.individualFactory = individualFactory;
		this.mutate = mutate;
		this.copy = copy;
		this.selector = selector;
		this.population = population;
	}

	public void initialize() throws TerminationException {
		for (int i = 0; i < OFFSIZE + POPSIZE; i++) {
			population.add(individualFactory.create());
		}
	}

	public void next() throws TerminationException {
		Collection lames = selector.getLames(OFFSIZE, population);
		population.removeAll(lames);

		Collection parents = selector.getParents(OFFSIZE, population);

		for (Individual parent : parents) {
			Genotype genotype = copy.copy(parent.getGenotype());
			mutate.mutate(genotype, 0.1);

			Individual child = individualFactory.create(genotype);
			population.add(child);
		}
	}
}