Update to use machine learning package

Use new machine learning package.