Stratifying drug treatment of cognitive impairments after traumatic brain injury using neuroimaging.