MOEAFramework-3.7.examples.Example2 Maven / Gradle / Ivy

Go to download
/* Copyright 2009-2023 David Hadka
 *
 * This file is part of the MOEA Framework.
 *
 * The MOEA Framework is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * The MOEA Framework is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with the MOEA Framework.  If not, see .
 */
import java.io.IOException;

import org.moeaframework.algorithm.NSGAII;
import org.moeaframework.core.NondominatedPopulation;
import org.moeaframework.core.PopulationIO;
import org.moeaframework.core.Problem;
import org.moeaframework.core.indicator.GenerationalDistance;
import org.moeaframework.core.indicator.Hypervolume;
import org.moeaframework.problem.DTLZ.DTLZ2;

/**
 * Quality indicators are used to compare results between different algorithms.  Here,
 * we calculate the hypervolume and generational distance relative to a reference set.
 * 
 * These reference sets contain optimal solutions to the problem.  Reference sets for
 * most test problems can be found in the ./pf/ directory.
 */
public class Example2 {

	public static void main(String[] args) throws IOException {
		// solve the 2-D DTLZ2 problem
		Problem problem = new DTLZ2(2);
		
		NSGAII algorithm = new NSGAII(problem);
		algorithm.run(10000);
		
		NondominatedPopulation result = algorithm.getResult();
		
		// load the reference set and evaluate the quality indicators
		NondominatedPopulation referenceSet = PopulationIO.readReferenceSet("pf/DTLZ2.2D.pf");
		Hypervolume hypervolume = new Hypervolume(problem, referenceSet);
		GenerationalDistance gd = new GenerationalDistance(problem, referenceSet);
		
		System.out.println("Hypervolume: " + hypervolume.evaluate(result));
		System.out.println("GD: " + gd.evaluate(result));
	}

}