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" }!! .find { (word, count) -> word == "\$numDocs" }!!
.second.toInt() .second.toInt()
val progressBarFactory: ProgressBarFactory by inject() val progressBarFactory: ProgressBarFactory by inject()
var step: Long = 0;
return progressBarFactory.new("compute idf", histogram.size.toLong()).use { progess -> return progressBarFactory.new("compute idf", histogram.size.toLong()).use { progess ->
csvWriter().openAsync(willProduce(resource.path).toFile(), append = false) { csvWriter().openAsync(willProduce(resource.path).toFile(), append = false) {
writeRow("word", "idf") writeRow("word", "idf")
histogram.forEach { (word, count) -> histogram.forEach { (word, count) ->
writeRow(word, idf(numDocs, count)) writeRow(word, idf(numDocs, count))
progess.update(step++) progess.step()
} }
} }
resource.path.toFile() resource.path.toFile()