use progress.step
parent
3e5534f184
commit
c40ab54012
|
|
@ -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()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue