Description
This model maps clinical entities and concepts to ICD11 codes to ICD10 codes.
How to use
document_assembler = DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")
doc2chunk = Doc2Chunk()\
.setInputCols(["document"])\
.setOutputCol("ner_chunk")
mapper = ChunkMapperModel.pretrained("icd10_icd11_mapper", "en", "clinical/models")\
.setInputCols(["ner_chunk"])\
.setOutputCol("mappings")
pipeline = Pipeline().setStages([
document_assembler,
doc2chunk,
mapper
])
data = spark.createDataFrame([["""[["D50.0"],["F51.0"],["D50.0"],["I10"]]"""]]).toDF("text")
result = pipeline.fit(data).transform(data)
document_assembler = nlp.DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")
doc2chunk = nlp.Doc2Chunk()\
.setInputCols(["document"])\
.setOutputCol("ner_chunk")
mapper = nlp.ChunkMapperModel.pretrained("icd10_icd11_mapper", "en", "clinical/models")\
.setInputCols(["ner_chunk"])\
.setOutputCol("mappings")
pipeline = nlp.Pipeline().setStages([
document_assembler,
doc2chunk,
mapper
])
data = spark.createDataFrame([["""[["D50.0"],["F51.0"],["D50.0"],["I10"]]"""]]).toDF("text")
result = pipeline.fit(data).transform(data)
val document_assembler = new DocumentAssembler()
.setInputCol("text")
.setOutputCol("document")
val doc2chunk = new Doc2Chunk()
.setInputCols(Array("document"))
.setOutputCol("ner_chunk")
val mapper = ChunkMapperModel.pretrained("icd10_icd11_mapper", "en", "clinical/models")
.setInputCols(Array("ner_chunk"))
.setOutputCol("mappings")
val pipeline = new Pipeline().setStages(Array(
document_assembler,
doc2chunk,
mapper
))
val data = Seq(Array"D50.0","F51.0","D50.0","I10")).toDF("text")
val result = pipeline.fit(data).transform(data)
Results
"
+----------+----------+
|icd10_code|icd11_code|
+----------+----------+
|D50.0 |[3A00.0Z] |
|F51.0 |[7A01] |
|D50.0 |[3A00.0Z] |
|I10 |[BA00.Z] |
+----------+----------+
Model Information
| Model Name: | icd10_icd11_mapper |
| Compatibility: | Healthcare NLP 6.4.1+ |
| License: | Licensed |
| Edition: | Official |
| Input Labels: | [ner_chunk] |
| Output Labels: | [mappings] |
| Language: | en |
| Size: | 141.5 KB |