Stagenet: Stage-aware neural networks for health risk prediction