Conditional out-of-distribution prediction - View it on GitHub