This code should get you ~100%:

https://gist.github.com/jamescalam/a4e2c46777f3a4035a58ae3fcd68c649

But looking at the error you are getting I assume there is something else wrong - maybe try training the model using Spyder/Pycharm rather than Jupyter (although I don't think it should make much difference).