dot-partition function

Partition the test set based on a trained tree