use progress.step

develop
Timo Bryant 2023-12-21 23:40:51 +01:00
parent 3e5534f184
commit c40ab54012
1 changed files with 1 additions and 4 deletions

View File

@ -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()