Refactored NERModel to test on data with multiple entity types.