step    0 loss=0.250
step  100 loss=0.166
step  200 loss=0.174
step  300 loss=0.172
step  400 loss=0.001
step  500 loss=0.000
step  600 loss=0.000
step  700 loss=0.000
step  800 loss=0.000
step  900 loss=0.000

Predictions:
[[0.   0.   0.01]
 [0.   1.   0.99]
 [1.   0.   0.99]
 [1.   1.   0.01]]