Creating a baseline machine learning pipeline