diff --git a/libraries/tfidf/src/main/kotlin/de/itkl/tfidf/InverseDocumentFrequency.kt b/libraries/tfidf/src/main/kotlin/de/itkl/tfidf/InverseDocumentFrequency.kt index 6b73d94..1e71705 100644 --- a/libraries/tfidf/src/main/kotlin/de/itkl/tfidf/InverseDocumentFrequency.kt +++ b/libraries/tfidf/src/main/kotlin/de/itkl/tfidf/InverseDocumentFrequency.kt @@ -29,15 +29,12 @@ class InverseDocumentFrequency : FileProcessor, KoinComponent { .find { (word, count) -> word == "\$numDocs" }!! .second.toInt() val progressBarFactory: ProgressBarFactory by inject() - - - var step: Long = 0; return progressBarFactory.new("compute idf", histogram.size.toLong()).use { progess -> csvWriter().openAsync(willProduce(resource.path).toFile(), append = false) { writeRow("word", "idf") histogram.forEach { (word, count) -> writeRow(word, idf(numDocs, count)) - progess.update(step++) + progess.step() } } resource.path.toFile()