A clinical goal for radiation treatment of a patient is set. A dose prediction model 150 is selected from a number of dose prediction models based on the clinical goal. A radiation treatment plan is then generated for the patient using the dose prediction model that was selected based on the clinical goal.