Great!
So partially I was wrong that I thought you that have some coding problem and partially correct because model was training just fine and only issue was the way you test.
But in your first post you did scale your input data X. Am I wrong?