Fix a bug that caused the retrieval of the document similarities to fail

loehnertz · Nov 30, 2019 · 76cffed · 76cffed
1 parent 9676d7a
commit 76cffed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/kotlin/SemanticCouplingCalculator.kt b/src/main/kotlin/SemanticCouplingCalculator.kt
@@ -23,9 +23,10 @@ class SemanticCouplingCalculator(private val files: Map<String, String>, private
     private lateinit var corpus: Corpus
 
     fun calculate() = runBlocking {
+        documentSimilarities.clear()
         corpus = Corpus(files.entries.mapConcurrently { parseFile(fileName = it.key, fileContents = it.value) }.toMutableSet())
         corpus = TfIdfCalculator(corpus, if (useLsi) null else fileSimilaritiesToCalculate).calculateForAllTerms()
-        SimilarityCalculator(corpus, fileSimilaritiesToCalculate, useLsi, numberOfLsiDimensions, maxLsiEpochs).calculateDocumentSimilarities().forEach { documentSimilarities.add(it) }.also { documentSimilarities.clear() }
+        SimilarityCalculator(corpus, fileSimilaritiesToCalculate, useLsi, numberOfLsiDimensions, maxLsiEpochs).calculateDocumentSimilarities().forEach { documentSimilarities.add(it) }
     }
 
     fun retrieveSimilaritiesAsListOfTriples(): List<Triple<String, String, Double>> {