i am stuck here any one help me to solve this problem. i think here is problem which i marked…
i can not see argmax part. Could you post some error message ?