Building Feature Extraction with Machine Learning