[XGBoost4J] Probability of predictions xgboost4j

[malakeel]

I have trained a model in python, and successfully loaded it from java.
However, I am unable to get the probability from java. I am just getting a single value.

clf = xgb.XGBClassifier( 
            max_depth = 32,
            objective='multi:softmax',
            n_estimators=200,
            tree_method = 'hist' )
clf.save_model("model_file_name.json" )

To obtain probabilities:

clf  = xgb.XGBClassifier().load_model(model_file_name )
predictions = clf.predict(X) 
prob = clf.predict_proba(X)

And it works fine in python. My java code is a long the line:

DMatrix mat = toDMatrix(numbers);
this.model = XGBoost.loadModel(modelPath);
float[][] prediction = this.model.predict(mat) ; 

I was not able to find a way to obtain the probability of the prediction in java. Am I missing something ?