I also thought on this problem and find out a possible solution: keep similar amount of data with different classes.