Vik/make package

Make machine learning into a package and update docs to reflect this.