training, by a prediction system, a first disease prediction machine learning (ML) model based on geographical disease data indicating new reported cases of a contagious disease within a first geographical area;

receiving, by the prediction system and from a medical facility, individual treatment data indicating dialysis treatment information associated with a patient undergoing an initial dialysis treatment, wherein the medical facility is associated with the first geographical area;

receiving, by the prediction system and from a blood testing laboratory, individual lab data indicating blood analysis information associated with the patient;

determining, by the prediction system, disease analysis results for the patient based on inputting the individual treatment data and the individual lab data into the first disease prediction ML model, wherein the disease analysis results indicate a likelihood of the patient being infected with the contagious disease;

providing, by the prediction system and to the medical facility, instructions indicating one or more responsive actions based on the disease analysis results, wherein the one or more responsive actions comprise lowering an ultrafiltration rate for the patient to use for a subsequent dialysis treatment;

performing, by a dialysis machine located at the medical facility, the subsequent dialysis treatment for the patient using the lowered ultrafiltration rate;

receiving, by the prediction system and from the medical facility, feedback information indicating a plurality of inaccuracy counts associated with the disease analysis results; and

based on comparing the plurality of inaccuracy counts with one or more thresholds, re-training, by the prediction system, the first disease prediction ML model.