tlatkowski / multihead-siamese-nets

Implementation of Siamese Neural Networks built upon multihead attention mechanism for text semantic similarity task.
MIT License
182 stars 43 forks source link

Predict only 2 results #23

Open longsamaa opened 4 years ago

longsamaa commented 4 years ago

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 commented 4 years ago

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

Regards