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("icd11_icd10_mapper", "en", "clinical/models")\
.setInputCols(["ner_chunk"])\
.setOutputCol("mappings")
pipeline = Pipeline().setStages([
document_assembler,
doc2chunk,
mapper
])
data = spark.createDataFrame([["""[["3A00.01"],["7A00"],["3A00.01"],["BA00.0"]]"""]]).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("icd11_icd10_mapper", "en", "clinical/models")\
.setInputCols(["ner_chunk"])\
.setOutputCol("mappings")
pipeline = nlp.Pipeline().setStages([
document_assembler,
doc2chunk,
mapper
])
data = spark.createDataFrame([["""[["3A00.01"],["7A00"],["3A00.01"],["BA00.0"]]"""]]).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("icd11_icd10_mapper","en","clinical/models")
.setInputCols(Array("ner_chunk"))
.setOutputCol("mappings")
val pipeline = new Pipeline().setStages(Array(
document_assembler,
doc2chunk,
mapper
))
val data = Seq(Array("3A00.01","7A00","3A00.01","BA00.0")).toDF("text")
val result = pipeline.fit(data).transform(data)
Results
"
+----------+----------+
|icd11_code|icd10_code|
+----------+----------+
|3A00.01 |[D50.0] |
|7A00 |[F51.0] |
|3A00.01 |[D50.0] |
|BA00.0 |[I10] |
+----------+----------+
Model Information
| Model Name: | icd11_icd10_mapper |
| Compatibility: | Healthcare NLP 6.4.1+ |
| License: | Licensed |
| Edition: | Official |
| Input Labels: | [ner_chunk] |
| Output Labels: | [mappings] |
| Language: | en |
| Size: | 213.7 KB |