Make pos num and neg num equal for the training set