Machine learning for classification of stratified geology from MWD data