Feature Request: Nested Geographical Data

For data similar to radon: https://www.tensorflow.org/datasets/catalog/radon

It would be useful to be able to inform the model of the hierarchical nature of geographic units (e.g. Country, State, County, Zip).