Distributional Random Forests: Heterogeneity Adjustment and Multivariate Distributional Regression

05/29/2020

∙

by   Domagoj Ćevid, et al.

∙

0

∙

share

We propose an adaptation of the Random Forest algorithm to estimate the conditional distribution of a possibly multivariate response. We suggest a new splitting criterion based on the MMD two-sample test, which is suitable for detecting heterogeneity in multivariate distributions. The weights provided by the forest can be conveniently used as an input to other methods in order to locally solve various learning problems. The code is available as R-package drf.