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.