Как создать дерево классификации с большим количеством разветвлений?(python)

При делении дерево развитвляется всего на 2 части, а нужно чтоб дерево разветвлялось на количество уникальных значений которое может принять столбец. Например столбец с названием "age" может принять значения [10, 20, 30] нужно чтоб разделение было такое (age==10, age==20, age==30). Обычное дерево разделяет примерно так (age>10)