Hugging Face's logo

general-forecast
/
kumo-tabular-medium-classifier

Safetensors
gfi
tabular
in-context-learning
kumo-tabular
Model card Files
xet
 Community
kumo-tabular-medium-classifier / README.md
tejas-generalforecast's picture
tejas-generalforecast
Convert medium/classifier.pt from nvidia/Kumo-Tabular
f7c3ce9 verified
preview code
|
Raw History Blame Contribute Delete
1.04 kB
metadata 
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 at revision 3c3e10bbdb590ace29e7026847f92db3c603096d), converted to safetensors for 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