Not available if prediction_only is set to true