argmax.java

//FUNCTION TO COMPUTE THE MAXIMUM PREDICTION AND ITS CONFIDENCE
    public Object[] argmax(float[] array){


        int best = -1;
        float best_confidence = 0.0f;

        for(int i = 0;i < array.length;i++){

            float value = array[i];

            if (value > best_confidence){

                best_confidence = value;
                best = i;
            }
        }

        return new Object[]{best,best_confidence};


    }

Hi,
This result came up when I was looking for any existing argmax in Java.
But this code only works for positive numbers right? If the array has zero or any negative numbers, it fails.
Also it doesn't check if the passed array is null or not.