Does `ClassIndex` mean which column is the "Label" (e.g. "iris-setosa", "iris-viriginica" in iris)? If that is the case, can we extend the `ClassIndex` from `int` to `[]int`? So we can have multi-label support.