| license: other | |
| license_name: openmdw-1.1 | |
| license_link: LICENSE | |
| base_model: nvidia/Kumo-Tabular | |
| library_name: gfi | |
| tags: | |
| - tabular | |
| - in-context-learning | |
| - kumo-tabular | |
| # Kumo Tabular medium classifier | |
| NVIDIA's Kumo Tabular medium classifier (`medium/classifier.pt` from | |
| [`nvidia/Kumo-Tabular`](https://huggingface.co/nvidia/Kumo-Tabular) at revision `3c3e10bbdb590ace29e7026847f92db3c603096d`), | |
| converted to safetensors for [gfi](https://github.com/generalforecast/gfi). The weights are | |
| unchanged. The network emits class probabilities over up to 10 classes. | |
| ## Files | |
| - `model.safetensors`: weights, under the parameter names of NVIDIA's PyTorch model. | |
| - `config.json`: `architecture_name` and `config`, the model's sizes as named in NVIDIA's | |
| `structured-data-models` (`MODEL_KWARGS["medium"]`). | |
| ## License | |
| OpenMDW 1.1, as for the original Kumo Tabular weights. NVIDIA's copyright notice and the license | |
| are kept in `LICENSE`. | |
| ## Source | |
| [NVIDIA structured-data-models](https://github.com/NVIDIA/structured-data-models) | |