ScriptSmith's picture
ONNX exports of TabICLv2, KumoTabular (small, medium, large), KumoRelational, Chronos-2 and Kumo-Anomaly for Tabular Data Workbench
b043bdf verified
Raw History Blame Contribute Delete
9.86 kB
{
"model": "TabICLv2",
"source": {
"package": "tabicl 2.2.0",
"repo": "jingang/TabICL",
"revision": "4dcd344ece2c00be9e831fdd35bed57b5ad83e19",
"checkpoints": {
"classifier": "tabicl-classifier-v2-20260212.ckpt",
"regressor": "tabicl-regressor-v2-20260212.ckpt"
},
"license": "BSD-3-Clause"
},
"opset": 18,
"exporter": "torch.onnx (dynamo) 2.14.0+cpu",
"limits": {
"maxClasses": 10,
"numQuantiles": 999,
"featureGroupSize": 3,
"cellDim": 128,
"rowDim": 512
},
"files": [
{
"task": "classifier",
"stage": "col",
"variant": "f16w",
"file": "classifier-col.f16w.onnx",
"bytes": 1894277,
"sha256": "a94e3998566acf9513e27ff38d5e8f7b9873bd8751d86192fd308ec426e7dd1c",
"cpuNodes": [],
"inputs": [
{
"name": "X",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
3
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"digits",
"train"
]
}
],
"outputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
]
},
{
"task": "classifier",
"stage": "icl",
"variant": "f16w",
"file": "classifier-icl.f16w.onnx",
"bytes": 52917443,
"sha256": "21791e2a7303c4d48611201f42f59f2ad69cf6ec5e6f528785234b889934e1a8",
"cpuNodes": [],
"inputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"train"
]
}
],
"outputs": [
{
"name": "out",
"type": "float32",
"shape": [
"batch",
"test",
10
]
}
]
},
{
"task": "classifier",
"stage": "row",
"variant": "f16w",
"file": "classifier-row.f16w.onnx",
"bytes": 855980,
"sha256": "177bffd02a588f58e25d0ed0269221214863109fc89a21966f1b9c9673bf8601",
"cpuNodes": [
"node_Concat_7"
],
"inputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
],
"outputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
}
]
},
{
"task": "classifier",
"stage": "col",
"variant": "fp32",
"file": "classifier-col.fp32.onnx",
"bytes": 3626570,
"sha256": "bb7221f89b5d73f4adc1b337bf0eaba7ab362c2494c33ce1f0f3549a2fa2b020",
"cpuNodes": [],
"inputs": [
{
"name": "X",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
3
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"digits",
"train"
]
}
],
"outputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
]
},
{
"task": "classifier",
"stage": "icl",
"variant": "fp32",
"file": "classifier-icl.fp32.onnx",
"bytes": 105366407,
"sha256": "7af95b18d1b45c15ebe26b89f0e066130ef6ee9adc4a3263bc8250f52b372260",
"cpuNodes": [],
"inputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"train"
]
}
],
"outputs": [
{
"name": "out",
"type": "float32",
"shape": [
"batch",
"test",
10
]
}
]
},
{
"task": "classifier",
"stage": "row",
"variant": "fp32",
"file": "classifier-row.fp32.onnx",
"bytes": 1641650,
"sha256": "316b5f0ce0aa2559731693aae9f126d941e4e1b01b9d9323e35df095b27be397",
"cpuNodes": [
"node_Concat_7"
],
"inputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
],
"outputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
}
]
},
{
"task": "regressor",
"stage": "col",
"variant": "f16w",
"file": "regressor-col.f16w.onnx",
"bytes": 1883270,
"sha256": "d434c1c37b5e0f53e8052869ba93682f42da4f3eb0f1154bda95b40ab6807436",
"cpuNodes": [],
"inputs": [
{
"name": "X",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
3
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"digits",
"train"
]
}
],
"outputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
]
},
{
"task": "regressor",
"stage": "icl",
"variant": "f16w",
"file": "regressor-icl.f16w.onnx",
"bytes": 54884074,
"sha256": "e2f25c737dee1524687c758487e85c38768984b5ec027ffb318da08fb54c454d",
"cpuNodes": [],
"inputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"train"
]
}
],
"outputs": [
{
"name": "out",
"type": "float32",
"shape": [
"batch",
"test",
999
]
}
]
},
{
"task": "regressor",
"stage": "row",
"variant": "f16w",
"file": "regressor-row.f16w.onnx",
"bytes": 851555,
"sha256": "280e72146067baaa2516ffa8483be2c368073fa4171f765a08ac2e8e4c801d8b",
"cpuNodes": [
"node_Concat_7"
],
"inputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
],
"outputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
}
]
},
{
"task": "regressor",
"stage": "col",
"variant": "fp32",
"file": "regressor-col.fp32.onnx",
"bytes": 3613048,
"sha256": "2ea2113646278399b1e58beb57d3e46d4496ee17e9c3871bafd4985dc92f4300",
"cpuNodes": [],
"inputs": [
{
"name": "X",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
3
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"digits",
"train"
]
}
],
"outputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
]
},
{
"task": "regressor",
"stage": "icl",
"variant": "fp32",
"file": "regressor-icl.fp32.onnx",
"bytes": 109348315,
"sha256": "17bb4ca71d88aea44b3f8886e376c41d382ded9ef71aafc594ae3b736a3b8916",
"cpuNodes": [],
"inputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
},
{
"name": "y",
"type": "float32",
"shape": [
"batch",
"train"
]
}
],
"outputs": [
{
"name": "out",
"type": "float32",
"shape": [
"batch",
"test",
999
]
}
]
},
{
"task": "regressor",
"stage": "row",
"variant": "fp32",
"file": "regressor-row.fp32.onnx",
"bytes": 1637225,
"sha256": "e03db9cdec62860e538afe61c6e1a6f7533a72756066da3660c508efb1f06559",
"cpuNodes": [
"node_Concat_7"
],
"inputs": [
{
"name": "E",
"type": "float32",
"shape": [
"batch",
"columns",
"rows",
128
]
}
],
"outputs": [
{
"name": "R",
"type": "float32",
"shape": [
"batch",
"rows",
512
]
}
]
}
]
}