Customized Machine Learning Class