I understand what the error message means but i just cant seem to find out where i have made such an error.
Ignore this query. I found my mistake.
Hello @Anish606! I am glad you found the error on your own.
For others, please use Numpy functions, instead of math. For example, math.log or math.exp is not recommended.