• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• sddf-master
  • src
    • main
      • resources
        • log4j.properties.template
        • project-default.properties
      • scala
        • de
          • unihamburg
            • vsis
              • sddf
                • print
                  • PipePrintHead.scala
                  • PipePrintHeadTuple.scala
                • writing
                  • PipeWriterTuple.scala
                  • PipeWriterTuplePairs.scala
                  • PipeWriterTupleClusterActualDate.scala
                  • ClusterWriterCsvFile.scala
                  • ResultWriterFile.scala
                  • TupleWriterFile.scala
                  • ClusterWriterCsvFileActualDate.scala
                  • PipeWriterTupleCluster.scala
                • sparkextensions
                  • RddSequenceFunctions.scala
                  • RddUtils.scala
                • shell
                  • ShellPrerequesites.scala
                  • MusicbrainzTemplate.scash
                  • BtwDemo.scash
                  • thesisExample.scash
                • examples
                  • PipeWordcount.scala
                • classification
                  • PipeClassificationSvm.scala
                  • PipePrintHeadFalsePositives.scala
                  • AbstractPipeClassification.scala
                  • AbstractPipePrintFalseTuples.scala
                  • PipeClassificationTrainingDataGenerator.scala
                  • PipePrintSampleFalseNegatives.scala
                  • PipeClassificationNaiveBayes.scala
                  • PipeClassificationDecisionTree.scala
                  • PipeAnalyseClassification.scala
                  • PipePrintSampleFalsePositives.scala
                  • PipePrintHeadFalseNegatives.scala
                  • PipeClassificationThreshold.scala
                  • PipeAnalyseClassificationTraining.scala
                • visualisation
                  • ModelRouterSilent.scala
                  • ModelRouter.scala
                  • Table.scala
                  • PipeSampler.scala
                  • model
                    • ClassificationModel.scala
                    • TrainingSetModel.scala
                    • RecallAndPrecision.scala
                    • GoldstandardModel.scala
                    • BlockingModel.scala
                    • BasicAnalysable.scala
                    • IndexingModelExtended.scala
                    • ReadingModel.scala
                    • GoldstandardClusterModel.scala
                    • Analysable.scala
                    • ClusterModel.scala
                    • AlgoAnalysable.scala
                    • IndexingModel.scala
                  • logger
                    • ClassificationOutputter.scala
                    • ReadingOutputter.scala
                    • GoldstandardOutputter.scala
                    • BlockingOutputter.scala
                    • BasicOutputter.scala
                    • AlgoOutputter.scala
                    • GoldstandardClusterOutputter.scala
                    • ModelRouterLogging.scala
                    • ClusterOutputter.scala
                    • Outputter.scala
                    • TrainingSetOutputter.scala
                    • IndexingOutputter.scala
                    • IndexingOutputterExtended.scala
                • clustering
                  • PipeAnalyseClustering.scala
                  • PipeClusteringTransitiveClosure.scala
                  • PipeClusteringStrongestPath.scala
                  • AbstractPipeClusteringGraph.scala
                • config
                  • Config.scala
                • indexing
                  • PipeIndexerSuffixArray.scala
                  • IndexingPipe.scala
                  • PipeIndexerSortedNeighborhood.scala
                  • PipeAnalyseIndexerExtended.scala
                  • PipeIndexerStandard.scala
                  • blocking
                    • PipeAnalyseBlocker.scala
                    • PipeBlockerStandard.scala
                    • PipeBlockerSuffixArray.scala
                    • keygeneration
                      • BlockingKeyBuilderBasic.scala
                      • BlockingKeyBuilder.scala
                    • PipeBlockerSortedNeighborhood.scala
                    • PipeBlockerPrefixArray.scala
                    • BlockingPipe.scala
                  • DummyIndexer.scala
                  • PipeAnalyseIndexer.scala
                • SddfContext.scala
                • convert
                  • PipeConvertClusterToPair.scala
                  • PipeConvertSetToPair.scala
                • Parameterized.scala
                • analyze
                  • PipeAnalyzeCount.scala
                • reading
                  • corpus
                    • AbstractPipeReaderTuple.scala
                    • PipePrintSampleCorpus.scala
                    • PipeStoreInContextCorpus.scala
                    • PipeAnalyseCorpus.scala
                    • PipePrintHeadCorpus.scala
                    • PipeReaderTupleFixedWidth.scala
                    • PipeContextReadCorpus.scala
                    • PipeReaderTupleCsv.scala
                  • TupleHashMap.scala
                  • TupleArray.scala
                  • PipeReaderOmitTail.scala
                  • IdConverterHex.scala
                  • SymPair.scala
                  • goldstandard
                    • PipePrintHeadGoldstandard.scala
                    • PipeReaderGoldstandardClusterOutput.scala
                    • PipeReaderGoldstandardIdsPairs.scala
                    • PipeAnalyseGoldstandard.scala
                    • PipeReaderGoldstandardIdToTuple.scala
                    • PipeStoreInContextGoldstandard.scala
                    • PipeReaderGoldstandard.scala
                    • PipePrintSampleGoldstandard.scala
                    • PipeReaderGoldstandardIdsCluster.scala
                    • PipeAnalyseGoldstandardCluster.scala
                  • PipeReaderOmitHead.scala
                  • FeatureIdNameMapping.scala
                  • IdConverter.scala
                  • IdConverterBasic.scala
                  • Tuple.scala
                • tuple
                  • TupleSymPairFunctions.scala
                • preprocessing
                  • PipePreprocessorReplaceRegex.scala
                  • TraitPipePreprocessor.scala
                  • PipePreprocessorMerger.scala
                  • PipePreprocessorSplitter.scala
                  • PipePreprocessorToLower.scala
                  • PipePreprocessorTrim.scala
                  • PipePreprocessorRemoveRegex.scala
                • tools
                  • ExactDuplicateFilter.scala
                  • ConvertGoldStandardPairFormat.scala
                • pipe
                  • Pipe.scala
                  • context
                    • ResultContext.scala
                    • AbstractPipeContext.scala
                    • GoldstandardContext.scala
                    • SddfPipeContext.scala
                    • CorpusContext.scala
                  • PipeOptimizer.scala
                  • PipeElementPassthrough.scala
                  • PipeMetaParallel.scala
                  • PipeElement.scala
                  • PipePassthrough.scala
                  • Pipeline.scala
                  • PipeFilterRDD.scala
                  • optimize
                    • PipeOptimizePersistAndName.scala
                    • PipeOptimizeUnpersist.scala
                    • PipeOptimizeCheckpoint.scala
                • filter
                  • PipeFilterDistinct.scala
                • SddfApp.scala
                • logging
                  • EachRunNewFileAppender.scala
                  • Logging.scala
                • similarity
                  • PipeSimilarity.scala
                  • SimilarityCalculator.scala
                  • aggregator
                    • Mean.scala
                    • SimilarityAggregator.scala
                    • Median.scala
                  • measures
                    • MeasureWrapperToLower.scala
                    • MeasureNumeric.scala
                    • MeasureEquality.scala
    • test
      • resources
        • log4j.properties.template
        • musicbrainz-1000.csv.dup
        • musicbrainz-10.csv.dup
        • project-default.properties
      • scala
        • de
          • unihamburg
            • vsis
              • sddf
                • test
                  • sparkextensions
                    • RddSequenceFunctionsTest.scala
                  • examples
                    • WordcountTest.scala
                  • classification
                    • PipeDecisionTest.scala
                    • PipeTrainingDataGeneratorTest.scala
                  • util
                    • FixtureHelper.scala
                    • MusicbrainzSchema.scala
                    • TestSddfPipeContext.scala
                    • LocalSparkContext.scala
                  • clustering
                    • StrongestPathClusteringTest.scala
                  • evaluation
                    • ClusterAnalyserTest.scala
                  • reading
                    • SymPairTest.scala
                    • PipeReaderOmitTailTest.scala
                    • goldstandard
                      • PipeGoldstandardReaderClusterTest.scala
                    • parser
                      • CsvTupleParserTest.scala
                      • FixedWidthTupleParserTest.scala
                  • blocking
                    • BlockingTest.scala
                    • keygeneration
                      • BlockingKeyBuilderBasicTest.scala
                    • SuffixArrayBlockingTest.scala
                    • SortedNeighbourhoodBlockerTest.scala
                  • logging
                    • LoggingTest.scala
                  • SparkApiTest.scala
  • launch.sh
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • README.md
  • build.sbt
  • .gitignore

package de.unihamburg.vsis.sddf.test.evaluation

import org.apache.spark.rdd.RDD
import org.scalatest.FunSuite

import de.unihamburg.vsis.sddf.reading.SymPair
import de.unihamburg.vsis.sddf.reading.Tuple
import de.unihamburg.vsis.sddf.test.util.FixtureHelper
import de.unihamburg.vsis.sddf.test.util.LocalSparkContext
import de.unihamburg.vsis.sddf.visualisation.model.ClusterModel

class ClusterAnalyserTest extends FunSuite with LocalSparkContext with FixtureHelper {
  test("Precission and recall test") {

    val analyser = new ClusterModel
    analyser.clusters = buildClusters()
    analyser.goldstandard = buildGoldstandard()

    assert(analyser.precision === 0.2857142857142857) // should be 2/7
    assert(analyser.recall === 0.6666666666666666) // should be 2/3

  }

  def buildClusters(): RDD[Set[Tuple]] = {
    val cluster1 = initializeTuples(0, 2).toSet
    val cluster2 = initializeTuples(3, 4).toSet
    val cluster3 = initializeTuples(5, 7).toSet

    sc.parallelize(Seq(cluster1, cluster2, cluster3))
  }

  def buildGoldstandard(): RDD[SymPair[Tuple]] = {
    val pair1 = createTuplePair(0, 1)
    val pair2 = createTuplePair(4, 7)
    val pair3 = createTuplePair(6, 7)

    sc.parallelize(Seq(pair1, pair2, pair3))
  }
  
}