Cleaner `predict` method