File size: 3,199 Bytes
b043bdf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | KumoTabular (small, medium and large; classifier and regressor) converted to
ONNX for Tabular Data Workbench.
This is an unofficial conversion. NVIDIA has not reviewed or endorsed it.
Kumo Tabular
Source: nvidia/Kumo-Tabular, tag v1.0.1
(revision 3c3e10bbdb590ace29e7026847f92db3c603096d)
https://huggingface.co/nvidia/Kumo-Tabular
Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
The weights are licensed under the OpenMDW License Agreement, version 1.1
(LICENSE).
Model code
The graphs were exported from the KumoTabular implementation in
structured-data-models 0.2.1 (https://github.com/NVIDIA/structured-data-models),
Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES, licensed under the
Apache License, Version 2.0 (LICENSE.structured-data-models where it is
distributed with these files). In that implementation the cell embedding is
derived from TabFM (https://github.com/google-research/tabfm, Copyright 2026
Google LLC, Apache License 2.0), and the attention scaling contains portions
derived from TabICLv2 (https://github.com/soda-inria/tabicl), whose licence
follows at the end of this file.
Changes
- <size>-<head>.f16w.onnx: the KumoTabular network (structured-data-models'
_KumoTabular.forward, without the preprocessing recipe or the ensemble)
exported with torch.onnx (dynamo, opset 20). The cell embedding's column
grouping, imputation and Fourier features are written without int64
arithmetic or scatters; what they compute is unchanged. Every weight of
1,024 or more elements is stored as float16 and cast back to float32 on
load; the other weights are unchanged.
TabICLv2
BSD 3-Clause License
Copyright (c) 2025, Soda team @ Inria
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright notice, this
list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
3. Neither the name of the copyright holder nor the names of its
contributors may be used to endorse or promote products derived from
this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|