CS726 Advanced ML IIT Bombay HW 3