Mapping SNOMED Codes with Their Corresponding ICDO Codes

Description

This model maps SNOMED codes to their corresponding ICDO 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("snomed_code")

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

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

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

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

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

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

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

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

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

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

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

val data = Seq("10013000").toDF("text")
val result = pipeline.fit(data).transform(data)

Results

|   snomed_code | icdo_code   | all_k_resolutions   |
|--------------:|:------------|:--------------------|
|      10013000 | C40.2       | C40.2:::            |
|     102291007 | C49.2       | C49.2:::C49.5       |
|     128501000 | C49.5       | C49.5:::C76.3       |

Model Information

Model Name: snomed_icdo_mapper_20260901
Compatibility: Healthcare NLP 6.4.1+
License: Licensed
Edition: Official
Input Labels: [snomed_code]
Output Labels: [mappings]
Language: en
Size: 270.2 KB