Predict only 2 results

[longsamaa]

Hello,

I have followed the instructions below:
https://github.com/tlatkowski/multihead-siamese-nets.
But when I predict, there are only 2 results like this:
[array ([[1.]], dtype = float32)]
[array ([[0.]], dtype = float32)]
Was my prediction result wrong?
Help me please

[tlatkowski]

hi @longsamaa
could you provide steps to reproduce your isses?

Regards