Revert "simplified predict"

This reverts commit 8d4f5169428cde64bb018132e0d0e7bc59ed3188.