AI model can help predict survival outcomes for patients with cancer