Changed to using scikit-learn for ML instead of hand-rolled neural networks