this ml model lets you to square the number on its own. you can change its function just by getting new sample dataset.