BNY interview question

Write the code for Supervised machine learning problem