[ML] module code cleanup pass

update scaladoc, separate data cleaning from model creation steps for better modularity, general improvements to code style and modularity and unit test coverage