An interpretable neural network for outcome prediction in traumatic brain injury