I am new to python and I can not understand where i need to make modifications on my code to make it work. I want to optimise a function that produce a prediction with two param