Kotlin/kotlindl

Update Dataset type to support multilabel datasets

PatrickLaflamme opened this issue · 1 comments

PatrickLaflamme commented

Currently, the Dataset abstract class only supports single-class datasets. This is because of the getY method defined here:

https://github.com/Kotlin/kotlindl/blob/master/dataset/src/jvmMain/kotlin/org/jetbrains/kotlinx/dl/dataset/Dataset.kt#L23

which only returns a single Float per index, instead of a FloatData.

zaleslaw commented

@PatrickLaflamme Could you give a case when it could be useful for KotlinDL? Did you try to do some cases?