search

Kotlin / kotlindl

High-level Deep Learning Framework written in Kotlin and inspired by Keras
Apache License 2.0
1.47k stars 103 forks source link

Update Dataset type to support multilabel datasets #578

Open PatrickLaflamme opened 1 year ago

PatrickLaflamme commented 1 year ago

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 11 months ago

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