diff options
Diffstat (limited to 'test/de/lmu/ifi/dbs/elki/evaluation/roc')
-rw-r--r-- | test/de/lmu/ifi/dbs/elki/evaluation/roc/TestComputeROC.java | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/test/de/lmu/ifi/dbs/elki/evaluation/roc/TestComputeROC.java b/test/de/lmu/ifi/dbs/elki/evaluation/roc/TestComputeROC.java new file mode 100644 index 00000000..206fbcfd --- /dev/null +++ b/test/de/lmu/ifi/dbs/elki/evaluation/roc/TestComputeROC.java @@ -0,0 +1,77 @@ +package de.lmu.ifi.dbs.elki.evaluation.roc; + +/* + This file is part of ELKI: + Environment for Developing KDD-Applications Supported by Index-Structures + + Copyright (C) 2012 + Ludwig-Maximilians-Universität München + Lehr- und Forschungseinheit für Datenbanksysteme + ELKI Development Team + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program 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 Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +import java.util.ArrayList; +import java.util.List; + +import junit.framework.Assert; + +import org.junit.Test; + +import de.lmu.ifi.dbs.elki.JUnit4Test; +import de.lmu.ifi.dbs.elki.database.ids.DBID; +import de.lmu.ifi.dbs.elki.database.ids.DBIDUtil; +import de.lmu.ifi.dbs.elki.database.ids.HashSetModifiableDBIDs; +import de.lmu.ifi.dbs.elki.evaluation.roc.ROC; +import de.lmu.ifi.dbs.elki.utilities.pairs.DoubleDoublePair; +import de.lmu.ifi.dbs.elki.utilities.pairs.Pair; + +/** + * Test to validate ROC curve computation. + * + * @author Erich Schubert + */ +public class TestComputeROC implements JUnit4Test { + /** + * Test ROC curve generation, including curve simplification + */ + @Test + public void testROCCurve() { + HashSetModifiableDBIDs positive = DBIDUtil.newHashSet(); + positive.add(DBIDUtil.importInteger(1)); + positive.add(DBIDUtil.importInteger(2)); + positive.add(DBIDUtil.importInteger(3)); + positive.add(DBIDUtil.importInteger(4)); + positive.add(DBIDUtil.importInteger(5)); + + ArrayList<Pair<Double, DBID>> distances = new ArrayList<Pair<Double, DBID>>(); + distances.add(new Pair<Double, DBID>(0.0, DBIDUtil.importInteger(1))); + distances.add(new Pair<Double, DBID>(1.0, DBIDUtil.importInteger(2))); + distances.add(new Pair<Double, DBID>(2.0, DBIDUtil.importInteger(6))); + distances.add(new Pair<Double, DBID>(3.0, DBIDUtil.importInteger(7))); + distances.add(new Pair<Double, DBID>(3.0, DBIDUtil.importInteger(3))); + distances.add(new Pair<Double, DBID>(4.0, DBIDUtil.importInteger(8))); + distances.add(new Pair<Double, DBID>(4.0, DBIDUtil.importInteger(4))); + distances.add(new Pair<Double, DBID>(5.0, DBIDUtil.importInteger(9))); + distances.add(new Pair<Double, DBID>(6.0, DBIDUtil.importInteger(5))); + + List<DoubleDoublePair> roccurve = ROC.materializeROC(9, positive, distances.iterator()); + // System.out.println(roccurve); + Assert.assertEquals("ROC curve too complex", 6, roccurve.size()); + + double auc = ROC.computeAUC(roccurve); + Assert.assertEquals("ROC AUC not right.", 0.6, auc, 0.0001); + } +} |