Systems and methods are provided herein for treating a patient in cardiogenic shock. An intravascular heart pump system is inserted into vasculature of the patient. The heart pump system has a cannula, pump outlet, pump inlet, and rotor. The heart pump system is positioned within the patient such that the cannula extends across the patient's aortic valve, the pump inlet is located within the patient's left ventricle, and the pump outlet is located within the patient's aorta. Data related to time-varying parameters of the heart pump system is acquired from the heart pump system. A plurality of features are extracted from the data. A probability of survival of the patient is determined based on the plurality of features and using a prediction model. The heart pump system is operated to treat the patient.