https://medium.com/codex/dealing-with-deprecation-in-machine-learning-libraries-a-step-by-step-guide-to-resolving-using-an-cb69ca6f4e10