Using machine learning to identify different types of brain injuries