Task prediction

[hailuu684]

Thank you for your great work. I would like to ask which part in your code that make the prediction of the task as it is mentioned in your paper? Thank you