Cleanup code style, formatting and basic refactoring in machine learning file

[juanis2112]

Conservative cleanup pass that applies AutoPEP8, performs additional manual cleanup, fixes imports and remove unnecessary ones, and resolves a few other minor issues.