Mapping ICDO Codes with Their Corresponding SNOMED Codes

Description

This model maps ICDO codes to their corresponding SNOMED codes.

It performs a direct lookup against the training dictionary, providing fast, exact-match code mapping.

Trained on SNOMED CT US Edition 20260901 crosswalk data.

Live Demo Open in Colab Copy S3 URI

How to use

documentAssembler = DocumentAssembler()\
    .setInputCol("text")\
    .setOutputCol("document")

chunkAssembler = Doc2Chunk()\
    .setInputCols(["document"])\
    .setOutputCol("icdo_code")

mapper = ChunkMapperModel.pretrained("icdo_snomed_mapper_20260901", "en", "clinical/models")\
    .setInputCols(["icdo_code"])\
    .setOutputCol("mappings")\
    .setRels(["snomed_code"])

pipeline = Pipeline(stages=[documentAssembler, chunkAssembler, mapper])

data = spark.createDataFrame([["8000/0"]]).toDF("text")
result = pipeline.fit(data).transform(data)
documentAssembler = nlp.DocumentAssembler()\
    .setInputCol("text")\
    .setOutputCol("document")

chunkAssembler = nlp.Doc2Chunk()\
    .setInputCols(["document"])\
    .setOutputCol("icdo_code")

mapper = medical.ChunkMapperModel.pretrained("icdo_snomed_mapper_20260901", "en", "clinical/models")\
    .setInputCols(["icdo_code"])\
    .setOutputCol("mappings")\
    .setRels(["snomed_code"])

pipeline = nlp.Pipeline(stages=[documentAssembler, chunkAssembler, mapper])

data = spark.createDataFrame([["8000/0"]]).toDF("text")
result = pipeline.fit(data).transform(data)

val documentAssembler = new DocumentAssembler()
    .setInputCol("text")
    .setOutputCol("document")

val chunkAssembler = new Doc2Chunk()
    .setInputCols(Array("document"))
    .setOutputCol("icdo_code")

val mapper = ChunkMapperModel.pretrained("icdo_snomed_mapper_20260901", "en", "clinical/models")
    .setInputCols(Array("icdo_code"))
    .setOutputCol("mappings")
    .setRels(Array("snomed_code"))

val pipeline = new Pipeline().setStages(Array(documentAssembler, chunkAssembler, mapper))

val data = Seq("8000/0").toDF("text")
val result = pipeline.fit(data).transform(data)

Results

| icdo_code   |   snomed_code | all_k_resolutions                            |
|:------------|--------------:|:---------------------------------------------|
| 8000/0      |       3898006 | 3898006:::                                   |
| 8000/1      |     400095002 | 400095002:::414389009:::783219005:::86251006 |
| 8170/3      |      25370001 | 25370001:::                                  |

Model Information

Model Name: icdo_snomed_mapper_20260901
Compatibility: Healthcare NLP 6.4.1+
License: Licensed
Edition: Official
Input Labels: [icdo_code]
Output Labels: [mappings]
Language: en
Size: 114.2 KB