How can i get prediction labels 0 or 1 for the test set.

[tramya28]

I am using MatchZoo version 2. I have several models from match zoo working. When i use the prediction, i get a prediction score between 0 and 1. Is there a way i could get 0 or 1 for each row in my test set instead of probability score.
The current code i use is

pred_x, pred_y = test_processed.unpack()
result = model.predict(pred_x)
When i look at the result, i see an array of probability scores = [[0.6524335 0.34756643]
[0.27325273 0.72674733]
[0.2813605 0.71863943]
...
[0.4537154 0.54628456]
[0.2675834 0.7324166 ]
[0.4178532 0.58214676]]

Is there a way i could get 0s and 1s instead of the probability scores for each test row?

[faneshion]

Typically, we can directly transform the score to labels through a pre-defined threshold, e.g., 0.5.