def load(): import medspacy nlp = medspacy.load(enable=["sentencizer", "tokenizer"]) # Add components from medspacy.target_matcher import TargetMatcher, TargetRule target_matcher = TargetMatcher(nlp) target_filepath = path.join(RESOURCES_DIR, "target_rules.json") target_rules = TargetRule.from_json(target_filepath) target_matcher.add(target_rules) nlp.add_pipe(target_matcher) from medspacy.context import ConTextComponent, ConTextRule context_filepath = path.join(RESOURCES_DIR, "context_rules.json") context = ConTextComponent(nlp, rules=None, add_attrs=CONTEXT_ATTRS) context_rules = ConTextRule.from_json(context_filepath) context.add(context_rules) nlp.add_pipe(context) from medspacy.section_detection import Sectionizer # TODO: Add radiology section rules sectionizer = Sectionizer(nlp) nlp.add_pipe(sectionizer) clf = DocumentClassifier(nlp) nlp.add_pipe(clf) return nlp
def test_from_json(self, from_json_file): assert ConTextRule.from_json(from_json_file)