['age', 'ejection_fraction', 'serum_creatinine', 'serum_sodium', 'time']