Yes, your own answer makes sense 
But you can do simpler : comparing the output of the sigmoid to 0.5 is equivalent to comparing the input of the sigmoid to 0 ! (see wikipedia 1)
So, you don’t need to call .sigmoid() ; just see where output < 0.