Correctly handles binary predictions now.