Probabilistic partition of unity networks for high-dimensional regression problems