BNY Interview Question

Write the code for Supervised machine learning problem