change the multiclass test to adapt it to xgBoost, change the get_s_b function to rebalance the weights