protected static int numLabels(INDArray labels) { FloatSet set = new FloatArraySet(); for (int i = 0; i < labels.size(0); i++) { set.add(labels.getFloat(i)); } return set.size(); }
protected static int numLabels(INDArray labels) { FloatSet set = new FloatArraySet(); for (int i = 0; i < labels.size(0); i++) { set.add(labels.getFloat(i)); } return set.size(); }