I have the following problem: Given two classes A and B which should be predicted using Sklearn or XGBoost. The function which has to be maximised is f(x) = A\'^2/(A\'+B\').