public INDArray confusionMatrix(INDArray labels, INDArray pred, INDArray weights,
      int numClasses) {
    NDValidation.validateNumerical("confusionMatrix", "labels", labels);
    NDValidation.validateNumerical("confusionMatrix", "pred", pred);
    NDValidation.validateNumerical("confusionMatrix", "weights", weights);