Create Balanced Partition (pyemma)