unify function

Unify tree-based model