TabImpute / matrixformer /artifacts.json
Tabimpute's picture
Release frozen MatrixFormer experts and inference examples
7275749 verified
Raw History Blame Contribute Delete
1.87 kB
{
"schema_version": 1,
"repo_id": "Tabimpute/TabImpute",
"source_repository": "https://github.com/jacobf18/tabular",
"source_revision": "2cf54ac47dbcc1154b35e875b527ed0c42a9f416",
"model_family": "new_stable",
"architecture": "Dwip v2-2",
"parameters_per_expert": 77998496,
"training_data": "Synthetic matrices only; no neural network training on real benchmark data",
"experts": {
"impute": {
"name": "MatrixFormer-Impute",
"checkpoint": "matrixformer/impute/best_validation_checkpoint.pth",
"checkpoint_bytes": 312100611,
"checkpoint_sha256": "c0ddf75274c8ff92ca5fd22c1263369ffa7e55eb7cb85fe80961b0afa101f7a9",
"config": "matrixformer/impute/config.json",
"config_sha256": "6981d286ef02b2c9261a8433ffb46a48a7fc68c65838e16895c6ad5888e7b07b",
"original_run": "mcpfn/runs/streamed_training/large_matrices_98k_20260930"
},
"recsys": {
"name": "Recommender specialist",
"checkpoint": "matrixformer/recsys/best_validation_checkpoint.pth",
"checkpoint_bytes": 312096259,
"checkpoint_sha256": "1fcadc0c5846f9c35fbbae4e3b519b9244b608272cb1eda68c0850132321f770",
"config": "matrixformer/recsys/config.json",
"config_sha256": "b9201e53abbb6f6c7cc1e1699d4fe8fead48208f28a979c32e46f40431a9f6be",
"original_run": "mcpfn/runs/recsys_goal_20260920/synthetic_user_scaled_parent_risk_400"
}
},
"inference": {
"prediction_stat": "mean",
"precision_mode": "amp_bfloat16",
"matrixformer_plus_policy": "observed_blend",
"example_normalization": "column",
"example_max_num_rows": 2000,
"example_max_num_cols": 1000,
"note": "The example is a generic matrix-completion recipe. Benchmark-specific normalization, clipping, orientation and context limits are documented in the source repository. MatrixFormer+ has no separate checkpoint."
}
}