Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=“mean”, without quantiles.
Id of the prediction, to get it again with get_predict_result.
Was this page helpful?
{"docsConfig":{"theme":"mint","$schema":"https://mintlify.com/docs.json","name":"Prior Labs","colors":{"primary":"#101075","light":"#ffffff","dark":"#1e293b"},"logo":{"light":"https://mintcdn.com/priorlabs/t197PY0EpQCr-YHY/logo/light.svg?fit=max\u0026auto=format\u0026n=t197PY0EpQCr-YHY\u0026q=85\u0026s=33643e275cfc08c99769e1719c4f53c2","dark":"https://mintcdn.com/priorlabs/t197PY0EpQCr-YHY/logo/light.svg?fit=max\u0026auto=format\u0026n=t197PY0EpQCr-YHY\u0026q=85\u0026s=33643e275cfc08c99769e1719c4f53c2"},"favicon":"https://mintcdn.com/priorlabs/SdCir_vgKw3mGacd/favicon.png?fit=max\u0026auto=format\u0026n=SdCir_vgKw3mGacd\u0026q=85\u0026s=88fa647a00e90147d312725e3354c544","appearance":{"default":"light","strict":true},"navbar":{"links":[{"href":"https://priorlabs.ai/contact?utm_source=docs","label":"Talk to Sales"}],"primary":{"type":"button","label":"Try TabPFN","href":"https://platform.priorlabs.ai/playground"}},"footer":{"socials":{"github":"https://github.com/priorlabs","linkedin":"https://www.linkedin.com/company/prior-labs"}},"fonts":{"heading":{"family":"Inter","weight":700},"body":{"family":"DM Sans","weight":400}},"integrations":{"ga4":{"measurementId":"G-9VZ68NQKPM"},"posthog":{"apiKey":"phc_GN5N25uDvJqsqnxmoM6Akr2CuyFIZsdu5ZokAXsPa1N","apiHost":"https://eu.i.posthog.com","sessionRecording":true}},"banner":{"content":"[New Thinking mode pushes the SOTA bar. Try it →](https://platform.priorlabs.ai/playground?tour=thinking)","dismissible":true},"interaction":{"drilldown":false},"markdown":{"instructions":["TabPFN-3 and later versions support up to 1,000,000 rows, subject to feature count and checkpoint/API limits.","For large datasets, use per-estimator subsampling, e.g. inference_config={\"SUBSAMPLE_SAMPLES\": 100_000}, in TabPFNClassifier or TabPFNRegressor from tabpfn or tabpfn_client.","Pass raw pandas DataFrames to fit and predict. Categorical strings/categories, raw text, and missing feature values are handled automatically; no manual integer/one-hot encoding, imputation, scaling, or outlier removal is needed."]},"styling":{"latex":true},"navigation":{"global":{"anchors":[{"anchor":"GitHub","icon":"github","href":"https://github.com/priorlabs/tabpfn"},{"anchor":"Discord","icon":"discord","href":"https://discord.com/invite/VJRuU3bSxt"},{"anchor":"Colab Demo","icon":"book","href":"https://colab.research.google.com/github/PriorLabs/TabPFN/blob/main/examples/notebooks/TabPFN_Demo_Local.ipynb"},{"anchor":"Changelog","icon":"clock-rotate-left","href":"/changelog"}]},"pages":[]}},"deployment":{"deploymentMetadataValue":{"subdomain":"priorlabs","hasAudioNarration":false,"actualSubdomain":"priorlabs","customDomains":["docs.priorlabs.ai"],"gitSource":{"type":"github","owner":"priorlabs","repo":"docs","deployBranch":"main","contentDirectory":"","isPrivate":true,"mountPath":"","label":""},"feedback":{"thumbs":true,"edits":false,"issues":false},"entitlements":{"AI_CHAT":{"status":"ENABLED"},"WEB_MCP":{"status":"ENABLED"},"NARRATION":{"status":"ENABLED"}},"buildId":"6aedf546fb71aa062f1155a4","clientVersion":"0.0.4088","searchFilterCounts":[{"language":"default","tag":"API reference","count":79},{"language":"default","tag":"Documentation","count":51},{"language":"default","tag":"Cookbook","count":26},{"language":"default","tag":"All Cookbooks","count":26},{"language":"default","tag":"tabpfn","count":25},{"language":"default","tag":"Capabilities","count":24},{"language":"default","tag":"tabpfn-extensions","count":23},{"language":"default","tag":"tabpfn-client","count":17},{"language":"default","tag":"TabPFNRegressor","count":10},{"language":"default","tag":"tabpfn-rel","count":8}]},"assistantConfig":{"enableStarterQuestions":false},"basePath":"","hasChatPermissions":true,"hasWebMcpPermissions":true,"toggles":[{"name":"kill-isr","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"agent-mintlify-slack","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"agent-integrations-settings","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"rbac-expansion","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"automations-pr-grouping","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"custom-prefix","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"analytics-v7","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"user-flows-analytics","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"no-publish-flow","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"workos-onboarding","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"llms-txt-v2","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"snapshot-serving-live","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"stripTemperatureMiddleware","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"support-platform-pylon-zendesk","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"snapshot-serving","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"instant-rollback","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"granular-notifications-server","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"granular-notifications","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"automations-fable","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"fixed-ai-action-pricing","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"automation-resolve-merge-conflict","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"scoped-publishing","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"mintql","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"page-level-analytics","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"editor-version-history-v2","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"billing-page-v2","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"full-page-search","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"workspace-projects","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"assistant-v5","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"fork-pr-previews","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"revalidation-jwt","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"workos-sso-scim","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"hobby-abuse-scan","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"dashboard-inbox","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"workos-signup","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"turbopuffer-dual-write","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"workos-enabled","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"approve-review-flow","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"editor-metadata","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"sso-setup-deadline-banner","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"turbopuffer-serve","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"live-agents","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false},{"name":"comment-mode","enabled":true,"variant":{"name":"disabled","enabled":false,"feature_enabled":true},"impressionData":false}],"googleFontsInlined":true},"customCss":[{"filePath":"style.css","content":".cookbook-filter-bar{display:flex;flex-direction:column;align-items:flex-start;gap:0.75rem;margin:1.5rem 0 2rem}.cookbook-filter-label{font-size:0.875rem;font-weight:600;color:#334155}.cookbook-tag-filters{display:flex;flex-wrap:wrap;gap:0.5rem;align-items:center}.cookbook-tag-filter-btn,.cookbook-tag-filter-clear{display:inline-flex;align-items:center;padding:0.375rem 0.75rem;border-radius:9999px;border:1px solid #cbd5e1;background:#ffffff;color:#334155;font-size:0.8125rem;font-weight:600;line-height:1.25rem;cursor:pointer;transition:background-color 0.15s ease, border-color 0.15s ease, color 0.15s ease}.cookbook-tag-filter-btn:hover,.cookbook-tag-filter-clear:hover{border-color:#101075;color:#101075}.cookbook-tag-filter-btn.is-active{background:#101075;border-color:#101075;color:#ffffff}.cookbook-tag-filter-clear{border-style:dashed}.cookbook-grid{display:grid;grid-template-columns:repeat(6, minmax(0, 1fr));gap:1rem}.cookbook-tile{height:100%;grid-column:span 2}.cookbook-tile>a,.cookbook-tile>div{height:100%}.cookbook-grid[data-featured-count=\"1\"] .cookbook-tile.is-featured{grid-column:span 6}.cookbook-grid[data-featured-count=\"2\"] .cookbook-tile.is-featured{grid-column:span 3}.cookbook-grid[data-featured-count=\"3\"] .cookbook-tile.is-featured{grid-column:span 2}@media (max-width: 1024px){.cookbook-grid{grid-template-columns:repeat(4, minmax(0, 1fr))}.cookbook-grid[data-featured-count=\"1\"] .cookbook-tile.is-featured{grid-column:span 4}.cookbook-grid[data-featured-count=\"2\"] .cookbook-tile.is-featured,.cookbook-grid[data-featured-count=\"3\"] .cookbook-tile.is-featured{grid-column:span 2}}@media (max-width: 640px){.cookbook-grid{grid-template-columns:minmax(0, 1fr)}.cookbook-tile,.cookbook-grid[data-featured-count] .cookbook-tile.is-featured{grid-column:1 / -1}}.cookbook-tile.is-featured>a,.cookbook-tile.is-featured>div{position:relative;border:2px solid #101075;box-shadow:0 8px 24px -12px rgba(16, 16, 117, 0.45);background-image:linear-gradient(\n 135deg,\n rgba(16, 16, 117, 0.06) 0%,\n rgba(16, 16, 117, 0) 55%\n )}.cookbook-tile.is-featured>a:hover{border-color:#101075;box-shadow:0 12px 32px -12px rgba(16, 16, 117, 0.6)}.cookbook-featured-badge{display:inline-flex;align-items:center;gap:0.25rem;margin:0 0 0.75rem;padding:0.125rem 0.625rem;border-radius:9999px;background:#101075;color:#ffffff;font-size:0.75rem;font-weight:700;letter-spacing:0.02em;line-height:1.25rem;text-transform:uppercase}.cookbook-tile.is-featured .cookbook-tile-description{font-size:1rem}.cookbook-tile-tags{display:flex;flex-wrap:wrap;gap:0.375rem;margin:0 0 0.75rem}.cookbook-tag-pill{display:inline-flex;align-items:center;padding:0.125rem 0.5rem;border-radius:9999px;background:#eef2ff;color:#101075;font-size:0.75rem;font-weight:600;line-height:1.25rem}.cookbook-tile-description{margin:0;color:#475569;font-size:0.9375rem;line-height:1.5;display:-webkit-box;-webkit-box-orient:vertical;-webkit-line-clamp:3;line-clamp:3;overflow:hidden;text-overflow:ellipsis}.cookbook-tile.is-featured .cookbook-tile-description{-webkit-line-clamp:2;line-clamp:2}.cookbook-meta{display:flex;flex-wrap:wrap;align-items:center;gap:0.75rem 1rem;margin:0 0 1.75rem}.cookbook-authors{margin:0;min-width:0}.cookbook-colab{display:flex;justify-content:flex-end;align-items:center;margin:0 0 0 auto}.cookbook-colab-button{display:inline-flex!important;align-items:center;gap:0.5rem;padding:0.5rem 1rem!important;font-family:Inter, ui-sans-serif, system-ui, sans-serif!important;font-size:0.875rem!important;font-weight:500!important;line-height:1.25!important;color:#101075!important;background-color:#ffffff!important;border:1px solid #101075!important;border-radius:0.75rem!important;text-decoration:none!important;box-shadow:none!important;transition:background-color 0.2s ease, border-color 0.2s ease, color 0.2s ease,\n transform 0.15s ease;cursor:pointer}.cookbook-colab-button:hover{background-color:#f3f3fa!important;border-color:#2a2a99!important;color:#2a2a99!important;transform:translateY(-1px)}.cookbook-colab-icon{width:1.125rem;height:1.125rem;flex-shrink:0;display:block}.cookbook-colab-label{white-space:nowrap}.cookbook-author-bar{display:inline-flex;flex-wrap:wrap;align-items:center;gap:0.625rem;padding:0.625rem 0.875rem;border:1px solid #e2e8f0;border-radius:9999px;background:linear-gradient(180deg, #ffffff 0%, #f8fafc 100%);box-shadow:0 1px 2px rgba(15, 23, 42, 0.04)}.cookbook-author-by{font-size:0.8125rem;font-weight:500;color:#64748b!important;letter-spacing:0.01em;flex-shrink:0}.cookbook-author-list{display:inline-flex;flex-wrap:wrap;align-items:center;gap:0.5rem 0.75rem}.cookbook-author-entry{display:inline-flex;align-items:center;gap:0.5rem}.cookbook-author-separator{color:#cbd5e1!important;font-size:1rem;line-height:1;user-select:none}.cookbook-author-name{font-size:0.9375rem;font-weight:600;color:#101075!important;white-space:nowrap}.cookbook-author-links{display:inline-flex;align-items:center;gap:0.375rem}.cookbook-author-icon-link{display:inline-flex;align-items:center;justify-content:center;width:1.75rem;height:1.75rem;border-radius:9999px;color:#475569!important;background:#ffffff;border:1px solid #e2e8f0;text-decoration:none;transition:color 0.15s ease, border-color 0.15s ease, background-color 0.15s ease, transform 0.15s ease}.cookbook-author-icon-link:hover{color:#101075!important;border-color:#c7d2fe;background:#eef2ff;transform:translateY(-1px)}.cookbook-author-icon{width:0.95rem;height:0.95rem;display:block}.compare-benchmark-table{overflow-x:auto;border:1px solid #dbe4f0;border-radius:0.75rem;background:#ffffff}.compare-task-switch-controls{display:inline-flex;gap:0.25rem;margin:0 0 1rem;padding:0.25rem;border:1px solid #dbe4f0;border-radius:0.5rem;background:#f8fafc}.compare-task-switch-button{padding:0.45rem 0.8rem;border:0;border-radius:0.35rem;background:transparent;color:#64748b;font-size:0.8125rem;font-weight:600;cursor:pointer}.compare-task-switch-button.is-active{background:#101075;color:#ffffff}.compare-task-panel[aria-hidden=\"true\"]{display:none}.compare-task-panel[aria-hidden=\"false\"]{display:block}.compare-benchmark-table table{display:table!important;width:100%!important;min-width:680px;margin:0;border-collapse:separate;border-spacing:0;table-layout:fixed!important}.compare-benchmark-table th:first-child,.compare-benchmark-table td:first-child{width:34%}.compare-benchmark-table th:not(:first-child),.compare-benchmark-table td:not(:first-child){width:13.2%}.compare-benchmark-table-performance table{min-width:0}.compare-benchmark-table-performance th:first-child,.compare-benchmark-table-performance td:first-child{width:50%}.compare-benchmark-table-performance th:not(:first-child),.compare-benchmark-table-performance td:not(:first-child){width:25%}.compare-benchmark-table-speed table{min-width:0}.compare-benchmark-table-speed th:first-child,.compare-benchmark-table-speed td:first-child{width:40%}.compare-benchmark-table-speed th:not(:first-child),.compare-benchmark-table-speed td:not(:first-child){width:20%}.compare-benchmark-table th,.compare-benchmark-table td{padding:0.75rem 1rem;border-bottom:1px solid #e8eef5;white-space:nowrap;text-align:right}.compare-benchmark-table th:first-child,.compare-benchmark-table td:first-child{text-align:left}.compare-benchmark-table th{color:#64748b;background:#f8fafc;font-size:0.75rem;font-weight:700}.compare-benchmark-table td{color:#475569;font-size:0.875rem}.benchmark-config-info{display:inline-block;position:relative;text-align:left;white-space:normal}.benchmark-config-trigger{padding:0;border:0;background:transparent;color:#12213a;cursor:pointer;text-decoration:underline dotted;text-underline-offset:0.2em}.benchmark-config-trigger:focus-visible{outline:2px solid #2855d9;outline-offset:2px}.benchmark-config-popover{position:fixed;z-index:1000;padding:0.75rem;border:1px solid #cbd5e1;border-radius:0.5rem;background:#ffffff;box-shadow:0 8px 24px rgb(15 23 42 / 18%);color:#334155;font-size:0.8125rem;line-height:1.5;text-align:left;white-space:normal}.benchmark-config-popover::before{position:absolute;top:-0.35rem;left:1rem;width:0.6rem;height:0.6rem;border-top:1px solid #cbd5e1;border-left:1px solid #cbd5e1;background:#ffffff;content:\"\";transform:rotate(45deg)}.benchmark-config-popover.is-above::before{top:auto;bottom:-0.35rem;border:0;border-right:1px solid #cbd5e1;border-bottom:1px solid #cbd5e1}.compare-data-types-table .data-type-supported,.compare-data-types-table .data-type-engineering{font-weight:600}.compare-data-types-table .data-type-supported{color:#101075}.compare-data-types-table .data-type-processed{color:#2855d9}.compare-data-types-table .data-type-engineering{color:#a21caf}.compare-benchmark-table tr:last-child td{border-bottom:0}.compare-benchmark-table .compare-table-row-tabpfn td{background:#eef4ff}.compare-benchmark-table .compare-table-row-tabpfn td:first-child{border-left:3px solid #2855d9}.compare-benchmark-table .compare-table-row-xgboost td{background:#fff5e9}.compare-benchmark-table .compare-table-row-xgboost td:first-child{border-left:3px solid #c85c0a}.compare-benchmark-table .compare-table-best{color:#12213a;font-weight:800}.compare-benchmark-note{margin:1rem 0 2rem;color:#64748b;font-size:0.8125rem;line-height:1.55}.compare-choice-grid{display:grid;grid-template-columns:repeat(2, minmax(0, 1fr));gap:1rem;margin:1.5rem 0 2.5rem}.compare-choice{padding:1.25rem;border:1px solid #dbe4f0;border-radius:0.5rem;background:#f8fbff}.compare-choice-competitor{border-color:#eadfce;background:#fdfaf5}.compare-choice h3{margin:0 0 0.75rem;font-size:0.95rem}.compare-choice ul{margin:0;padding-left:1.1rem;color:#475569;font-size:0.875rem}.compare-choice li{margin:0.5rem 0}@media (max-width: 700px){.compare-choice-grid{grid-template-columns:minmax(0, 1fr)}}.sidebar-group-title{font-size:0.5rem!important;font-weight:600!important}[data-testid=\"sidebar-group-title\"]{font-size:0.5rem!important;font-weight:600!important}.sidebar .group-title{font-size:0.5rem!important;font-weight:600!important}.sidebar-group-title{margin-bottom:0.5rem!important;margin-top:1rem!important}.table-container{width:auto!important;overflow-x:auto!important}.table-container table{width:auto!important;table-layout:auto!important}.table-container table th:first-child,.table-container table td:first-child{text-align:left!important;white-space:nowrap!important;padding-right:1rem!important}.table-container table th:last-child,.table-container table td:last-child{text-align:left!important}.nav-logo{width:27%!important;height:auto}h1,h2,h3,h4,h5,h6{color:#101075!important}.heading-1,.heading-2,.heading-3,.heading-4,.heading-5,.heading-6{color:#101075!important}[data-heading]{color:#101075!important}.heading{color:#101075!important}.main-content,.content,.page-content{background-color:#f8f8f9!important}body,p,div,a,li,td,th{color:#0c0c0c!important}strong,b{font-weight:700!important;color:#0c0c0c!important}.paragraph{color:#0c0c0c!important}a{color:#0c0c0c!important}.sidebar-group-title{font-size:0.5rem!important;font-weight:600!important}[data-testid=\"sidebar-group-title\"]{font-size:0.5rem!important;font-weight:600!important}.sidebar .group-title{font-size:0.5rem!important;font-weight:600!important}.sidebar-group-title{margin-bottom:0.5rem!important;margin-top:1rem!important}.table-container-dc3e57e5{width:100%!important;overflow-x:auto!important}.table-container-dc3e57e5 table{width:100%!important;table-layout:fixed!important}.table-container-dc3e57e5 table th:first-child,.table-container-dc3e57e5 table td:first-child{width:50%!important;text-align:left!important;white-space:nowrap!important;padding-right:1rem!important}.table-container-dc3e57e5 table th:nth-child(2),.table-container-dc3e57e5 table td:nth-child(2){width:50%!important;text-align:left!important}.table-container-nowrap{width:100%!important;overflow-x:auto!important}.table-container-nowrap table{width:auto!important;table-layout:auto!important}.table-container-nowrap table th,.table-container-nowrap table td{white-space:nowrap!important;text-align:left!important;padding-right:1rem!important}.table-container-nowrap table th:first-child,.table-container-nowrap table td:first-child{width:auto!important;padding-right:1.5rem!important}.metering-operations th,.metering-operations td{padding:0.75rem 1rem}.metering-operations thead th{font-weight:700}#banner{background:linear-gradient(90deg, #101075 0%, #1e3a8a 60%, #2563eb 100%)!important;font-size:0.875rem;letter-spacing:0.01em}#banner,#banner *{color:#ffffff!important}#banner strong{font-weight:700}#banner a{color:#ffffff!important;font-weight:600;text-decoration:underline;text-underline-offset:3px;text-decoration-color:rgba(255, 255, 255, 0.6);transition:text-decoration-color 0.15s ease}#banner a:hover{text-decoration-color:#ffffff}#navbar a[href=\"https://priorlabs.ai/contact?utm_source=docs\"]{display:inline-flex!important;align-items:center;gap:0.625rem;padding:0.5rem 1rem!important;font-size:0.875rem!important;font-weight:500!important;color:#101075!important;background-color:#ffffff!important;border:1px solid #101075!important;border-radius:0.75rem!important;text-decoration:none!important;transition:background-color 0.2s ease, border-color 0.2s ease, color 0.2s ease;cursor:pointer}#navbar a[href=\"https://priorlabs.ai/contact?utm_source=docs\"]::after{content:\"\";display:inline-block;width:6px;height:6px;border-right:1.5px solid #101075;border-bottom:1.5px solid #101075;transform:rotate(-45deg);margin-top:-1px;flex-shrink:0;transition:border-color 0.2s ease}#navbar a[href=\"https://priorlabs.ai/contact?utm_source=docs\"]:hover{background-color:#f3f3fa!important;border-color:#2a2a99!important;color:#2a2a99!important}#navbar a[href=\"https://priorlabs.ai/contact?utm_source=docs\"]:hover::after{border-right-color:#2a2a99;border-bottom-color:#2a2a99}.research-search-bar{display:flex;flex-direction:column;gap:0.5rem;margin:1.5rem 0 1rem}.research-search-label{font-size:0.875rem;font-weight:600;color:#334155}.research-search-input-wrap{position:relative;display:flex;align-items:center;width:100%;max-width:32rem}.research-search-icon{position:absolute;left:0.75rem;width:1.125rem;height:1.125rem;color:#94a3b8;pointer-events:none}.research-search-input{width:100%;padding:0.625rem 0.75rem 0.625rem 2.375rem;border:1px solid #cbd5e1;border-radius:0.75rem;background:#ffffff;color:#0f172a;font-size:0.9375rem;line-height:1.375rem;transition:border-color 0.15s ease, box-shadow 0.15s ease}.research-search-input::placeholder{color:#94a3b8}.research-search-input:focus{outline:none;border-color:#101075;box-shadow:0 0 0 3px rgba(16, 16, 117, 0.12)}.research-filter-bar{display:flex;flex-direction:column;align-items:flex-start;gap:0.75rem;margin:1.5rem 0 1rem}.research-filter-label{font-size:0.875rem;font-weight:600;color:#334155}.research-tag-filters{display:flex;flex-wrap:wrap;gap:0.5rem;align-items:center}.research-tag-filter-btn,.research-tag-filter-clear{display:inline-flex;align-items:center;padding:0.375rem 0.75rem;border-radius:9999px;border:1px solid #cbd5e1;background:#ffffff;color:#334155;font-size:0.8125rem;font-weight:600;line-height:1.25rem;cursor:pointer;transition:background-color 0.15s ease, border-color 0.15s ease, color 0.15s ease}.research-tag-filter-btn:hover,.research-tag-filter-clear:hover{border-color:#101075;color:#101075}.research-tag-filter-btn.is-active{background:#101075;border-color:#101075;color:#ffffff}.research-tag-filter-clear{border-style:dashed}.research-count{font-size:0.8125rem;font-weight:500;color:#64748b;margin:0 0 1rem}.research-grid{display:grid;grid-template-columns:repeat(2, minmax(0, 1fr));gap:1rem}@media (max-width: 900px){.research-grid{grid-template-columns:minmax(0, 1fr)}}.research-tile{display:flex!important;flex-direction:column;gap:0.75rem;padding:1.25rem 1.25rem 1rem;border:1px solid #e2e8f0;border-radius:0.75rem;background:#ffffff;color:inherit!important;text-decoration:none!important;transition:border-color 0.15s ease, box-shadow 0.15s ease, transform 0.15s ease;height:100%}.research-tile.is-hidden{display:none!important}.research-tile:hover{border-color:#101075;box-shadow:0 6px 20px rgba(16, 16, 117, 0.08);transform:translateY(-1px)}.research-tile-cta{background:linear-gradient(135deg, #eef2ff 0%, #f8fafc 100%);border-color:#c7d2fe;border-style:dashed}.research-tile-cta:hover{border-style:solid}.research-tile-cta .research-tile-meta{color:#101075;font-weight:600;text-transform:none;letter-spacing:0;font-size:0.8125rem}.research-tile-header{display:flex;flex-wrap:wrap;gap:0.375rem}.research-tag-pill{display:inline-flex;align-items:center;padding:0.125rem 0.5rem;border-radius:9999px;background:#eef2ff;color:#101075;font-size:0.75rem;font-weight:600;line-height:1.25rem}.research-tile-title{margin:0;font-size:1rem;font-weight:600;line-height:1.35;color:#0f172a!important}.research-tile-description{margin:0;color:#475569;font-size:0.9375rem;line-height:1.55;flex-grow:1}.research-tile-footer{display:flex;flex-direction:column;gap:0.25rem;margin-top:0.25rem;padding-top:0.75rem;border-top:1px solid #f1f5f9;font-size:0.8125rem;color:#64748b}.research-tile-authors{font-weight:500;color:#334155}.research-tile-meta{color:#94a3b8;font-size:0.75rem;text-transform:uppercase;letter-spacing:0.03em}.research-pagination{display:none;align-items:center;justify-content:center;gap:1rem;margin:1.5rem 0 0;width:100%;clear:both;grid-column:1 / -1}.research-pagination.is-visible{display:flex}.research-pagination-btn{display:inline-flex;align-items:center;justify-content:center;min-width:6rem;padding:0.5rem 0.875rem;border:1px solid #cbd5e1;border-radius:0.75rem;background:#ffffff;color:#334155;font-size:0.875rem;font-weight:600;line-height:1.25rem;cursor:pointer;transition:background-color 0.15s ease, border-color 0.15s ease, color 0.15s ease}.research-pagination-btn:hover:not(:disabled){border-color:#101075;color:#101075}.research-pagination-btn:disabled{cursor:not-allowed;opacity:0.45}.research-pagination-status{font-size:0.875rem;font-weight:600;color:#475569}.python-reference-table table{width:100%!important;min-width:640px;table-layout:fixed!important}.python-reference-table th,.python-reference-table td{white-space:normal!important;overflow-wrap:anywhere;vertical-align:top}.python-reference-table th:nth-child(1){width:20%}.python-reference-table th:nth-child(2){width:20%}.python-reference-table th:nth-child(3){width:16%}.python-reference-table th:nth-child(4){width:44%}.python-reference-table code{white-space:normal;overflow-wrap:anywhere}.python-reference-table code.python-reference-parameter{overflow-wrap:normal;word-break:normal}.python-reference-table [id]{scroll-margin-top:180px}.python-reference-returns table{min-width:0}.python-reference-returns th:nth-child(1){width:32%}.python-reference-returns th:nth-child(2){width:68%}body:has(.python-reference-table) #pagination{gap:1rem}body:has(.python-reference-table) #pagination>a{min-width:0;max-width:48%}body:has(.python-reference-table) #pagination>a span{overflow-wrap:anywhere}body:has(.python-reference-table) #pagination>a svg{flex-shrink:0}.python-reference-heading{display:flex;align-items:baseline;justify-content:space-between;flex-wrap:wrap;gap:0.75rem 1rem;margin:2.5rem 0 1rem}.python-reference-heading h2{margin:0!important;min-width:0}.python-reference-heading h2 code{font-size:0.85em;white-space:normal;overflow-wrap:anywhere}.python-reference-source{display:inline-flex;align-items:center;gap:0.375rem;flex-shrink:0;padding:0.25rem 0.5rem;border:1px solid color-mix(in srgb, currentColor 20%, transparent);border-radius:0.375rem;color:inherit!important;font-size:0.75rem;line-height:1.25rem;text-decoration:none!important}.python-reference-source:hover{background:color-mix(in srgb, currentColor 6%, transparent)}.python-reference-type a{font:inherit;padding:0;border-radius:0;background:transparent}"}],"page":{"pagePath":"api-reference/prediction/predict-tabpfn-json-api","slug":"/api-reference/prediction/predict-tabpfn-json-api","pageMetadata":{"title":"Predict (TabPFN JSON API)","description":"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\"mean\", without quantiles.","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/predict","href":"/api-reference/prediction/predict-tabpfn-json-api","autogeneratedByOpenApi":true},"description":{"compiledSource":"\"use strict\";\nconst {jsx: _jsx} = arguments[0];\nconst {useMDXComponents: _provideComponents} = arguments[0];\nfunction _createMdxContent(props) {\n const _components = {\n p: \"p\",\n ..._provideComponents(),\n ...props.components\n };\n return _jsx(_components.p, {\n children: \"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=“mean”, without quantiles.\"\n });\n}\nfunction MDXContent(props = {}) {\n const {wrapper: MDXLayout} = {\n ..._provideComponents(),\n ...props.components\n };\n return MDXLayout ? _jsx(MDXLayout, {\n ...props,\n children: _jsx(_createMdxContent, {\n ...props\n })\n }) : _createMdxContent(props);\n}\nreturn {\n default: MDXContent\n};\n","frontmatter":{},"scope":{}},"pageType":"default","mdx":{"mdxSource":{"compiledSource":"\"use strict\";const{Fragment:_Fragment,jsx:_jsx}=arguments[0];const{useMDXComponents:_provideComponents}=arguments[0];function _createMdxContent(props){return _jsx(_Fragment,{})}function MDXContent(props={}){const{wrapper:MDXLayout}={..._provideComponents(),...props.components};return MDXLayout?_jsx(MDXLayout,{...props,children:_jsx(_createMdxContent,{...props})}):_createMdxContent(props)}return{default:MDXContent}","frontmatter":{},"scope":{"config":{},"pageMetadata":{"title":"Predict (TabPFN JSON API)","description":"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\"mean\", without quantiles.","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/predict","href":"/api-reference/prediction/predict-tabpfn-json-api","autogeneratedByOpenApi":true}}},"mdxSourceWithNoJs":null},"mdxExtracts":{"tableOfContents":[],"codeExamples":{}},"jsLinks":["/_mintlify/custom/priorlabs/dc47dfb9d8e46daa/cookieyes.js","/_mintlify/custom/priorlabs/d8ce73b7f7a5de7b/posthog-consent.js","/_mintlify/custom/priorlabs/bd145ba8c5f78a12/cookbook-filter.js","/_mintlify/custom/priorlabs/4fffc985e576f41c/research-filter.js","/_mintlify/custom/priorlabs/09720b48fd3a180c/tabarena-switch.js"],"openApiReferenceData":{"operation":{"summary":"Predict (TabPFN JSON API)","description":"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\"mean\", without quantiles.","operationId":"tabpfn_predict","tags":["Prediction"],"responses":{"200":"fd3ac1d1-4d12-448a-a8f1-da32543782ca","404":"114dbbf4-2444-44b9-b4d3-2c44dd0659b5","409":"8c96247c-08bb-4010-8bef-2e3772607147","default":"2172b909-eeed-4264-a126-543eaeaf7d5f"},"requestBody":"448e4fde-4c15-4f7d-aaee-a64e930f4a45","method":"post","type":"path","servers":["13114613-4d51-455e-9bff-41c0cd6fa421"],"security":["dbc4483c-2d19-44ca-8290-c974a9870b1b"],"path":"/tabpfn/predict","baseUrlOptions":["https://api.priorlabs.ai"],"title":"Predict (TabPFN JSON API)","codeSamples":[],"requiredOnlyExamples":false,"requestExampleType":"generated","responseExampleType":"generated"},"dependencies":{"parameters":{},"requestBody":{"required":true,"content":{"application/json":{"schema":{"oneOf":[{"type":"object","title":"PredictRequest","required":["fitted_train_set_id","task_config","test_set_upload_id"],"properties":{"fitted_train_set_id":{"format":"uuid","title":"Fitted Train Set Id","type":"string","isRequired":true,"uniqueKey":"oneOf.0.fitted_train_set_id","typeLabel":"string\u003cuuid>"},"task_config":{"oneOf":[{"type":"object","title":"ClassifierConfig","properties":{"task":{"const":"classification","default":"classification","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.0.task_config.oneOf.0.task","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.model_path","typeLabel":"string | null"},"balance_probabilities":{"oneOf":[{"type":"boolean","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.balance_probabilities.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.balance_probabilities.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config.balance_probabilities","typeLabel":"boolean | null"}},"title":"ClassifierTabPFNConfig","type":"object","description":"Model settings for the fit.","uniqueKey":"oneOf.0.task_config.oneOf.0.tabpfn_config","typeLabel":"ClassifierTabPFNConfig · object"},"predict_params":{"additionalProperties":false,"properties":{"output_type":{"enum":["probas","preds","top_k"],"title":"ClassifierOutputType","type":"string","default":"probas","description":"What `prediction` contains:\n\n* `probas`: one probability for each class, for each row.\n* `preds`: one label for each row.\n* `top_k`: the most probable classes for each row.","uniqueKey":"oneOf.0.task_config.oneOf.0.predict_params.output_type","typeLabel":"enum\u003cstring>"},"top_k":{"oneOf":[{"type":"integer","title":"Top K","description":"Number of classes that `top_k` returns for each row. Required with `output_type=top_k`. Other output types do not permit it.","minimum":1,"uniqueKey":"oneOf.0.task_config.oneOf.0.predict_params.top_k.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Top K","description":"Number of classes that `top_k` returns for each row. Required with `output_type=top_k`. Other output types do not permit it.","uniqueKey":"oneOf.0.task_config.oneOf.0.predict_params.top_k.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.0.predict_params.top_k","typeLabel":"integer | null"}},"title":"ClassifierPredictParams","type":"object","description":"What the prediction returns.","uniqueKey":"oneOf.0.task_config.oneOf.0.predict_params","typeLabel":"ClassifierPredictParams · object"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.0.task_config.oneOf.0","typeLabel":"ClassifierConfig · object"},{"type":"object","title":"RegressorConfig","properties":{"task":{"const":"regression","default":"regression","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.0.task_config.oneOf.1.task","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config.model_path","typeLabel":"string | null"}},"title":"RegressorTabPFNConfig","type":"object","description":"Model settings for the fit.","uniqueKey":"oneOf.0.task_config.oneOf.1.tabpfn_config","typeLabel":"RegressorTabPFNConfig · object"},"predict_params":{"additionalProperties":false,"properties":{"output_type":{"enum":["mean","median","mode","quantiles","full","main"],"title":"RegressorOutputType","type":"string","default":"mean","description":"What `prediction` contains:\n\n* `mean`, `median` or `mode`: one value for each row.\n* `quantiles`: one list for each quantile.\n* `main`: the mean, median, mode and quantiles.\n* `full`: the full predicted distribution. It accepts fewer rows.","uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params.output_type","typeLabel":"enum\u003cstring>"},"quantiles":{"oneOf":[{"type":"array","items":{"maximum":1,"minimum":0,"type":"number","uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params.quantiles.oneOf.0.items.[INDEX]","typeLabel":"number"},"title":"Quantiles","description":"Quantile levels between 0 and 1, for `quantiles` and `main`. If unset, TabPFN uses its default list of 9 levels.","uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params.quantiles.oneOf.0","typeLabel":"number[]"},{"type":"null","title":"Quantiles","description":"Quantile levels between 0 and 1, for `quantiles` and `main`. If unset, TabPFN uses its default list of 9 levels.","uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params.quantiles.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params.quantiles","typeLabel":"number[] | null"}},"title":"RegressorPredictParams","type":"object","description":"What the prediction returns.","uniqueKey":"oneOf.0.task_config.oneOf.1.predict_params","typeLabel":"RegressorPredictParams · object"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.0.task_config.oneOf.1","typeLabel":"RegressorConfig · object"}],"isRequired":true,"uniqueKey":"oneOf.0.task_config","typeLabel":"ClassifierConfig · object | RegressorConfig · object"},"test_set_upload_id":{"format":"uuid","title":"Test Set Upload Id","type":"string","isRequired":true,"uniqueKey":"oneOf.0.test_set_upload_id","typeLabel":"string\u003cuuid>"}},"additionalProperties":false,"uniqueKey":"oneOf.0","typeLabel":"PredictRequest · object"},{"type":"object","title":"InlinePredictRequest","required":["fitted_train_set_id","task_config","x_test_parquet_base64"],"properties":{"fitted_train_set_id":{"format":"uuid","title":"Fitted Train Set Id","type":"string","isRequired":true,"uniqueKey":"oneOf.1.fitted_train_set_id","typeLabel":"string\u003cuuid>"},"task_config":{"oneOf":[{"type":"object","title":"ClassifierConfig","properties":{"task":{"const":"classification","default":"classification","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.1.task_config.oneOf.0.task","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.model_path","typeLabel":"string | null"},"balance_probabilities":{"oneOf":[{"type":"boolean","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.balance_probabilities.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.balance_probabilities.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config.balance_probabilities","typeLabel":"boolean | null"}},"title":"ClassifierTabPFNConfig","type":"object","description":"Model settings for the fit.","uniqueKey":"oneOf.1.task_config.oneOf.0.tabpfn_config","typeLabel":"ClassifierTabPFNConfig · object"},"predict_params":{"additionalProperties":false,"properties":{"output_type":{"enum":["probas","preds","top_k"],"title":"ClassifierOutputType","type":"string","default":"probas","description":"What `prediction` contains:\n\n* `probas`: one probability for each class, for each row.\n* `preds`: one label for each row.\n* `top_k`: the most probable classes for each row.","uniqueKey":"oneOf.1.task_config.oneOf.0.predict_params.output_type","typeLabel":"enum\u003cstring>"},"top_k":{"oneOf":[{"type":"integer","title":"Top K","description":"Number of classes that `top_k` returns for each row. Required with `output_type=top_k`. Other output types do not permit it.","minimum":1,"uniqueKey":"oneOf.1.task_config.oneOf.0.predict_params.top_k.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Top K","description":"Number of classes that `top_k` returns for each row. Required with `output_type=top_k`. Other output types do not permit it.","uniqueKey":"oneOf.1.task_config.oneOf.0.predict_params.top_k.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.0.predict_params.top_k","typeLabel":"integer | null"}},"title":"ClassifierPredictParams","type":"object","description":"What the prediction returns.","uniqueKey":"oneOf.1.task_config.oneOf.0.predict_params","typeLabel":"ClassifierPredictParams · object"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.1.task_config.oneOf.0","typeLabel":"ClassifierConfig · object"},{"type":"object","title":"RegressorConfig","properties":{"task":{"const":"regression","default":"regression","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.1.task_config.oneOf.1.task","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config.model_path","typeLabel":"string | null"}},"title":"RegressorTabPFNConfig","type":"object","description":"Model settings for the fit.","uniqueKey":"oneOf.1.task_config.oneOf.1.tabpfn_config","typeLabel":"RegressorTabPFNConfig · object"},"predict_params":{"additionalProperties":false,"properties":{"output_type":{"enum":["mean","median","mode","quantiles","full","main"],"title":"RegressorOutputType","type":"string","default":"mean","description":"What `prediction` contains:\n\n* `mean`, `median` or `mode`: one value for each row.\n* `quantiles`: one list for each quantile.\n* `main`: the mean, median, mode and quantiles.\n* `full`: the full predicted distribution. It accepts fewer rows.","uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params.output_type","typeLabel":"enum\u003cstring>"},"quantiles":{"oneOf":[{"type":"array","items":{"maximum":1,"minimum":0,"type":"number","uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params.quantiles.oneOf.0.items.[INDEX]","typeLabel":"number"},"title":"Quantiles","description":"Quantile levels between 0 and 1, for `quantiles` and `main`. If unset, TabPFN uses its default list of 9 levels.","uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params.quantiles.oneOf.0","typeLabel":"number[]"},{"type":"null","title":"Quantiles","description":"Quantile levels between 0 and 1, for `quantiles` and `main`. If unset, TabPFN uses its default list of 9 levels.","uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params.quantiles.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params.quantiles","typeLabel":"number[] | null"}},"title":"RegressorPredictParams","type":"object","description":"What the prediction returns.","uniqueKey":"oneOf.1.task_config.oneOf.1.predict_params","typeLabel":"RegressorPredictParams · object"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.1.task_config.oneOf.1","typeLabel":"RegressorConfig · object"}],"isRequired":true,"uniqueKey":"oneOf.1.task_config","typeLabel":"ClassifierConfig · object | RegressorConfig · object"},"x_test_parquet_base64":{"description":"The test set as a base64 Parquet file, in place of `test_set_upload_id`. The decoded file can be at most `inline_test_set_max_parquet_bytes` from `get_settings`. If that value is 0, the server does not accept inline test sets.","title":"X Test Parquet Base64","type":"string","isRequired":true,"uniqueKey":"oneOf.1.x_test_parquet_base64","typeLabel":"string"}},"additionalProperties":false,"uniqueKey":"oneOf.1","typeLabel":"InlinePredictRequest · object"}],"uniqueKey":"","typeLabel":"PredictRequest · object | InlinePredictRequest · object"},"example":{"test_set_upload_id":"33333333-3333-4333-8333-333333333333","fitted_train_set_id":"22222222-2222-4222-8222-222222222222","task_config":{"task":"classification","predict_params":{"output_type":"probas"}}},"examples":{"Example":{"title":"Example","value":{"test_set_upload_id":"33333333-3333-4333-8333-333333333333","fitted_train_set_id":"22222222-2222-4222-8222-222222222222","task_config":{"task":"classification","predict_params":{"output_type":"probas"}}}}},"exampleType":"examples"}},"exampleType":"examples"},"responses":{"fd3ac1d1-4d12-448a-a8f1-da32543782ca":{"description":"Completed prediction or a pending job. For a pending job, poll /tabpfn/get_predict_result with fitted_test_set_id.","content":{"application/json":{"schema":{"oneOf":[{"type":"object","title":"PredictResponse","required":["prediction","metadata"],"properties":{"prediction":{"oneOf":[{"type":"array","items":{"oneOf":[{}],"uniqueKey":"oneOf.0.prediction.oneOf.0.items.[INDEX]","typeLabel":"any"},"title":"Prediction","isRequired":true,"uniqueKey":"oneOf.0.prediction.oneOf.0","typeLabel":"any[]"},{"type":"array","items":{"items":{"oneOf":[{}],"uniqueKey":"oneOf.0.prediction.oneOf.1.items.[INDEX].items.[INDEX]","typeLabel":"any"},"type":"array","uniqueKey":"oneOf.0.prediction.oneOf.1.items.[INDEX]","typeLabel":"any[]"},"title":"Prediction","isRequired":true,"uniqueKey":"oneOf.0.prediction.oneOf.1","typeLabel":"any[][]"},{"type":"object","title":"Prediction","additionalProperties":{"oneOf":[{"type":"array","items":{"oneOf":[{}],"uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties.oneOf.0.items.[INDEX]","typeLabel":"any"},"uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties.oneOf.0","typeLabel":"any[]"},{"type":"array","items":{"items":{"oneOf":[{}],"uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties.oneOf.1.items.[INDEX].items.[INDEX]","typeLabel":"any"},"type":"array","uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties.oneOf.1.items.[INDEX]","typeLabel":"any[]"},"uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties.oneOf.1","typeLabel":"any[][]"}],"uniqueKey":"oneOf.0.prediction.oneOf.2.additionalProperties","typeLabel":"any[] | any[][]"},"isRequired":true,"uniqueKey":"oneOf.0.prediction.oneOf.2","typeLabel":"Prediction · object"}],"isRequired":true,"uniqueKey":"oneOf.0.prediction","typeLabel":"any[] | any[][] | Prediction · object"},"metadata":{"oneOf":[{"type":"object","title":"ClassifierMetadata","required":["test_set_num_rows","test_set_num_cols","package_version","tabpfn_config"],"properties":{"test_set_num_rows":{"description":"Number of rows that the model predicted.","title":"Test Set Num Rows","type":"integer","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.0.test_set_num_rows","typeLabel":"integer"},"test_set_num_cols":{"description":"Number of columns after encoding. This number can be different from the number of columns that you sent.","title":"Test Set Num Cols","type":"integer","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.0.test_set_num_cols","typeLabel":"integer"},"package_version":{"description":"Version of the TabPFN package that made the prediction.","title":"Package Version","type":"string","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.0.package_version","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.model_path","typeLabel":"string | null"},"balance_probabilities":{"oneOf":[{"type":"boolean","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.balance_probabilities.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Balance Probabilities","description":"If true, the model adjusts the probabilities for the class distribution of the training data. This can help with imbalanced classes when the metric ignores class imbalance, such as balanced accuracy.","uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.balance_probabilities.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config.balance_probabilities","typeLabel":"boolean | null"}},"title":"ClassifierTabPFNConfig","type":"object","description":"The model settings that the request used.","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.0.tabpfn_config","typeLabel":"ClassifierTabPFNConfig · object"},"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size that the request used. Null when no ensemble ran.","uniqueKey":"oneOf.0.metadata.oneOf.0.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size that the request used. Null when no ensemble ran.","uniqueKey":"oneOf.0.metadata.oneOf.0.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.n_estimators","typeLabel":"integer | null"},"billing_model_version":{"oneOf":[{"type":"string","title":"Billing Model Version","description":"The model version for the usage count of the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.0.billing_model_version.oneOf.0","typeLabel":"string"},{"type":"null","title":"Billing Model Version","description":"The model version for the usage count of the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.0.billing_model_version.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.billing_model_version","typeLabel":"string | null"},"execution_mode":{"oneOf":[{"type":"string","title":"Execution Mode","description":"How the server ran the request. Null when not reported.","enum":["standard","thinking","cache"],"uniqueKey":"oneOf.0.metadata.oneOf.0.execution_mode.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Execution Mode","description":"How the server ran the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.0.execution_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.execution_mode","typeLabel":"enum\u003cstring> | null"},"cache_outcome":{"oneOf":[{"type":"string","title":"Cache Outcome","description":"Whether the request used a kept fit. Null when not reported.","enum":["not_requested","hit","miss","fallback"],"uniqueKey":"oneOf.0.metadata.oneOf.0.cache_outcome.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Cache Outcome","description":"Whether the request used a kept fit. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.0.cache_outcome.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.cache_outcome","typeLabel":"enum\u003cstring> | null"},"task":{"const":"classification","default":"classification","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.0.metadata.oneOf.0.task","typeLabel":"string"},"classes":{"oneOf":[{"type":"array","items":{"oneOf":[{"type":"string","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0.items.[INDEX].oneOf.0","typeLabel":"string"},{"type":"integer","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0.items.[INDEX].oneOf.1","typeLabel":"integer"},{"type":"number","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0.items.[INDEX].oneOf.2","typeLabel":"number"},{"type":"boolean","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0.items.[INDEX].oneOf.3","typeLabel":"boolean"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0.items.[INDEX]","typeLabel":"string | integer | number | boolean"},"title":"Classes","description":"Class labels in ascending order. They name the columns of `probas` and the labels of `preds`. Null with `top_k`, because `top_k` has its own `prediction.classes`.","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.0","typeLabel":"(string | integer | number | boolean)[]"},{"type":"null","title":"Classes","description":"Class labels in ascending order. They name the columns of `probas` and the labels of `preds`. Null with `top_k`, because `top_k` has its own `prediction.classes`.","uniqueKey":"oneOf.0.metadata.oneOf.0.classes.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.classes","typeLabel":"(string | integer | number | boolean)[] | null"},"top_k":{"oneOf":[{"type":"integer","title":"Top K","description":"The `top_k` value of the request.","uniqueKey":"oneOf.0.metadata.oneOf.0.top_k.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Top K","description":"The `top_k` value of the request.","uniqueKey":"oneOf.0.metadata.oneOf.0.top_k.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.0.top_k","typeLabel":"integer | null"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.0","typeLabel":"ClassifierMetadata · object"},{"type":"object","title":"RegressorMetadata","required":["test_set_num_rows","test_set_num_cols","package_version","tabpfn_config"],"properties":{"test_set_num_rows":{"description":"Number of rows that the model predicted.","title":"Test Set Num Rows","type":"integer","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.1.test_set_num_rows","typeLabel":"integer"},"test_set_num_cols":{"description":"Number of columns after encoding. This number can be different from the number of columns that you sent.","title":"Test Set Num Cols","type":"integer","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.1.test_set_num_cols","typeLabel":"integer"},"package_version":{"description":"Version of the TabPFN package that made the prediction.","title":"Package Version","type":"string","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.1.package_version","typeLabel":"string"},"tabpfn_config":{"additionalProperties":false,"properties":{"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","maximum":8,"minimum":1,"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size. The model averages this number of forward passes, and each pass uses a different preprocessed view of the data. If unset, TabPFN selects the size. It selects 8 for most data. For wide tables, it selects more, so that each column gets to at least one ensemble member.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.n_estimators","typeLabel":"integer | null"},"categorical_features_indices":{"oneOf":[{"type":"array","items":{"type":"integer","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0.items.[INDEX]","typeLabel":"integer"},"title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.0","typeLabel":"integer[]"},{"type":"null","title":"Categorical Features Indices","description":"0-based indices of the categorical columns. If unset, the model finds the categorical columns.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.categorical_features_indices.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.categorical_features_indices","typeLabel":"integer[] | null"},"softmax_temperature":{"oneOf":[{"type":"number","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.softmax_temperature.oneOf.0","typeLabel":"number"},{"type":"null","title":"Softmax Temperature","description":"A lower value makes the predicted probabilities more confident. The value 1.0 has no effect. If unset, the model uses its default value.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.softmax_temperature.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.softmax_temperature","typeLabel":"number | null"},"average_before_softmax":{"oneOf":[{"type":"boolean","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.average_before_softmax.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Average Before Softmax","description":"If true, the model averages the logits of the ensemble members before the softmax. If false, it averages their probabilities after the softmax.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.average_before_softmax.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.average_before_softmax","typeLabel":"boolean | null"},"random_state":{"oneOf":[{"type":"integer","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.random_state.oneOf.0","typeLabel":"integer"},{"type":"null","title":"Random State","description":"Random seed. The same seed gives the same results. If unset, TabPFN uses its default seed.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.random_state.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.random_state","typeLabel":"integer | null"},"inference_config":{"oneOf":[{"type":"object","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","additionalProperties":true,"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_config.oneOf.0","typeLabel":"Inference Config · object"},{"type":"null","title":"Inference Config","description":"Advanced inference settings. See `tabpfn.inference_config.InferenceConfig` in the TabPFN package.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_config.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_config","typeLabel":"Inference Config · object | null"},"inference_precision":{"oneOf":[{"type":"string","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","enum":["autocast","auto"],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_precision.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Inference Precision","description":"`autocast` always uses mixed precision. `auto` selects the precision from the hardware.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_precision.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.inference_precision","typeLabel":"enum\u003cstring> | null"},"ignore_pretraining_limits":{"default":true,"description":"If true, the server accepts data outside the size ranges of the model's pre-training, and predictions on that data may be worse. If false, the server rejects that data.","title":"Ignore Pretraining Limits","type":"boolean","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.ignore_pretraining_limits","typeLabel":"boolean"},"n_preprocessing_jobs":{"default":4,"description":"Number of worker processes for preprocessing.","title":"N Preprocessing Jobs","type":"integer","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.n_preprocessing_jobs","typeLabel":"integer"},"fit_mode":{"enum":["low_memory","fit_preprocessors","fit_with_cache","batched"],"title":"FitMode","type":"string","default":"fit_preprocessors","description":"How the model prepares and keeps the fit. With `fit_with_cache`, the server keeps the fit, and later requests can use it again. Not all servers support `fit_with_cache`.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.fit_mode","typeLabel":"enum\u003cstring>"},"device":{"oneOf":[{"type":"array","items":{"type":"string","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.device.oneOf.0.items.[INDEX]","typeLabel":"string"},"title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","minItems":1,"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.device.oneOf.0","typeLabel":"string[]"},{"type":"null","title":"Device","description":"Devices for the request, such as `cuda:0`. If unset, the server selects the devices.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.device.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.device","typeLabel":"string[] | null"},"memory_saving_mode":{"oneOf":[{"type":"boolean","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.0","typeLabel":"boolean"},{"type":"null","title":"Memory Saving Mode","description":"If true, the model divides parts of the computation into batches to use less memory. If unset, the model decides.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.memory_saving_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.memory_saving_mode","typeLabel":"boolean | null"},"model_path":{"oneOf":[{"type":"string","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.model_path.oneOf.0","typeLabel":"string"},{"type":"null","title":"Model Path","description":"The checkpoint, by file name or model name. If unset, the server uses its default checkpoint.","uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.model_path.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config.model_path","typeLabel":"string | null"}},"title":"RegressorTabPFNConfig","type":"object","description":"The model settings that the request used.","isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.1.tabpfn_config","typeLabel":"RegressorTabPFNConfig · object"},"n_estimators":{"oneOf":[{"type":"integer","title":"N Estimators","description":"Ensemble size that the request used. Null when no ensemble ran.","uniqueKey":"oneOf.0.metadata.oneOf.1.n_estimators.oneOf.0","typeLabel":"integer"},{"type":"null","title":"N Estimators","description":"Ensemble size that the request used. Null when no ensemble ran.","uniqueKey":"oneOf.0.metadata.oneOf.1.n_estimators.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.n_estimators","typeLabel":"integer | null"},"billing_model_version":{"oneOf":[{"type":"string","title":"Billing Model Version","description":"The model version for the usage count of the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.1.billing_model_version.oneOf.0","typeLabel":"string"},{"type":"null","title":"Billing Model Version","description":"The model version for the usage count of the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.1.billing_model_version.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.billing_model_version","typeLabel":"string | null"},"execution_mode":{"oneOf":[{"type":"string","title":"Execution Mode","description":"How the server ran the request. Null when not reported.","enum":["standard","thinking","cache"],"uniqueKey":"oneOf.0.metadata.oneOf.1.execution_mode.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Execution Mode","description":"How the server ran the request. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.1.execution_mode.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.execution_mode","typeLabel":"enum\u003cstring> | null"},"cache_outcome":{"oneOf":[{"type":"string","title":"Cache Outcome","description":"Whether the request used a kept fit. Null when not reported.","enum":["not_requested","hit","miss","fallback"],"uniqueKey":"oneOf.0.metadata.oneOf.1.cache_outcome.oneOf.0","typeLabel":"enum\u003cstring>"},{"type":"null","title":"Cache Outcome","description":"Whether the request used a kept fit. Null when not reported.","uniqueKey":"oneOf.0.metadata.oneOf.1.cache_outcome.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.metadata.oneOf.1.cache_outcome","typeLabel":"enum\u003cstring> | null"},"task":{"const":"regression","default":"regression","description":"The type of prediction.","title":"Task","type":"string","uniqueKey":"oneOf.0.metadata.oneOf.1.task","typeLabel":"string"}},"additionalProperties":false,"isRequired":true,"uniqueKey":"oneOf.0.metadata.oneOf.1","typeLabel":"RegressorMetadata · object"}],"isRequired":true,"uniqueKey":"oneOf.0.metadata","typeLabel":"ClassifierMetadata · object | RegressorMetadata · object"},"timings":{"oneOf":[{"type":"object","title":"PredictTimings","description":"Where a predict's time went, per stage. Queue wait covers everything\noutside worker processing: waiting for a free worker and transport.","properties":{"test_set_transform_queue_wait_s":{"oneOf":[{"type":"number","title":"Test Set Transform Queue Wait S","description":"Seconds the test set transform waited for a worker.","uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_queue_wait_s.oneOf.0","typeLabel":"number"},{"type":"null","title":"Test Set Transform Queue Wait S","description":"Seconds the test set transform waited for a worker.","uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_queue_wait_s.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_queue_wait_s","typeLabel":"number | null"},"test_set_transform_s":{"oneOf":[{"type":"number","title":"Test Set Transform S","description":"Seconds a worker spent loading, validating and preprocessing the test set.","uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_s.oneOf.0","typeLabel":"number"},{"type":"null","title":"Test Set Transform S","description":"Seconds a worker spent loading, validating and preprocessing the test set.","uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_s.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.timings.oneOf.0.test_set_transform_s","typeLabel":"number | null"},"predict_queue_wait_s":{"oneOf":[{"type":"number","title":"Predict Queue Wait S","description":"Seconds the prediction waited for a worker.","uniqueKey":"oneOf.0.timings.oneOf.0.predict_queue_wait_s.oneOf.0","typeLabel":"number"},{"type":"null","title":"Predict Queue Wait S","description":"Seconds the prediction waited for a worker.","uniqueKey":"oneOf.0.timings.oneOf.0.predict_queue_wait_s.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.timings.oneOf.0.predict_queue_wait_s","typeLabel":"number | null"},"predict_s":{"oneOf":[{"type":"number","title":"Predict S","description":"Seconds a worker spent predicting.","uniqueKey":"oneOf.0.timings.oneOf.0.predict_s.oneOf.0","typeLabel":"number"},{"type":"null","title":"Predict S","description":"Seconds a worker spent predicting.","uniqueKey":"oneOf.0.timings.oneOf.0.predict_s.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.timings.oneOf.0.predict_s","typeLabel":"number | null"}},"additionalProperties":false,"uniqueKey":"oneOf.0.timings.oneOf.0","typeLabel":"PredictTimings · object"},{"type":"null","uniqueKey":"oneOf.0.timings.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.timings","typeLabel":"PredictTimings · object | null"},"fitted_test_set_id":{"oneOf":[{"type":"string","title":"Fitted Test Set Id","description":"Id of the prediction, to get it again with get_predict_result.","format":"uuid","uniqueKey":"oneOf.0.fitted_test_set_id.oneOf.0","typeLabel":"string\u003cuuid>"},{"type":"null","title":"Fitted Test Set Id","description":"Id of the prediction, to get it again with get_predict_result.","uniqueKey":"oneOf.0.fitted_test_set_id.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.fitted_test_set_id","typeLabel":"string\u003cuuid> | null"}},"additionalProperties":false,"uniqueKey":"oneOf.0","typeLabel":"PredictResponse · object"},{"type":"object","title":"PredictPendingResponse","description":"A predict still running when the wait for it ended.","required":["fitted_test_set_id","retry_in_secs"],"properties":{"fitted_test_set_id":{"format":"uuid","title":"Fitted Test Set Id","type":"string","isRequired":true,"uniqueKey":"oneOf.1.fitted_test_set_id","typeLabel":"string\u003cuuid>"},"retry_in_secs":{"title":"Retry In Secs","type":"number","isRequired":true,"uniqueKey":"oneOf.1.retry_in_secs","typeLabel":"number"},"status":{"description":"Status of a predict job.","enum":["pending","completed","failed"],"title":"PredictStatus","type":"string","default":"pending","uniqueKey":"oneOf.1.status","typeLabel":"enum\u003cstring>"}},"additionalProperties":false,"uniqueKey":"oneOf.1","typeLabel":"PredictPendingResponse · object"}],"uniqueKey":"","typeLabel":"PredictResponse · object | PredictPendingResponse · object"},"examples":{"Example":{"title":"Example","value":{"prediction":["\u003cunknown>"],"metadata":{"test_set_num_rows":123,"test_set_num_cols":123,"package_version":"\u003cstring>","tabpfn_config":{"n_estimators":4,"categorical_features_indices":[123],"softmax_temperature":123,"average_before_softmax":true,"random_state":123,"inference_config":{},"inference_precision":"autocast","ignore_pretraining_limits":true,"n_preprocessing_jobs":4,"fit_mode":"fit_preprocessors","device":["\u003cstring>"],"memory_saving_mode":true,"model_path":"\u003cstring>","balance_probabilities":true},"n_estimators":123,"billing_model_version":"\u003cstring>","execution_mode":"standard","cache_outcome":"not_requested","task":"classification","classes":["\u003cstring>"],"top_k":123},"timings":{"test_set_transform_queue_wait_s":123,"test_set_transform_s":123,"predict_queue_wait_s":123,"predict_s":123},"fitted_test_set_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a"}}},"exampleType":"generated"}},"exampleType":"generated"},"114dbbf4-2444-44b9-b4d3-2c44dd0659b5":{"description":"Fitted training set not found.","content":{"application/json":{"schema":{"additionalProperties":false,"properties":{"message":{"description":"What went wrong.","title":"Message","type":"string","isRequired":true,"uniqueKey":"message","typeLabel":"string"},"error_code":{"default":"NOT_FOUND","title":"Error Code","type":"string","x-sdk-keep-default":true,"uniqueKey":"error_code","typeLabel":"string"},"trace_id":{"oneOf":[{"type":"string","title":"Trace Id","description":"Id to quote when reporting the error.","format":"uuid","uniqueKey":"trace_id.oneOf.0","typeLabel":"string\u003cuuid>"},{"type":"null","title":"Trace Id","description":"Id to quote when reporting the error.","uniqueKey":"trace_id.oneOf.1","typeLabel":"null"}],"uniqueKey":"trace_id","typeLabel":"string\u003cuuid> | null"}},"required":["message"],"title":"NotFoundErrorResponse","type":"object","uniqueKey":"","typeLabel":"NotFoundErrorResponse · object"},"examples":{"Example":{"title":"Example","value":{"message":"\u003cstring>","error_code":"NOT_FOUND","trace_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a"}}},"exampleType":"generated"}},"exampleType":"generated"},"8c96247c-08bb-4010-8bef-2e3772607147":{"description":"Refit the model when error_code is NEEDS_REFITTING or KV_CACHE_INVALID.","content":{"application/json":{"schema":{"additionalProperties":false,"description":"A prediction that failed after the server created it. get_predict_result\nreports the same failure under its id.","properties":{"message":{"description":"What went wrong.","title":"Message","type":"string","isRequired":true,"uniqueKey":"message","typeLabel":"string"},"fitted_test_set_id":{"description":"Id of the prediction that failed.","format":"uuid","title":"Fitted Test Set Id","type":"string","isRequired":true,"uniqueKey":"fitted_test_set_id","typeLabel":"string\u003cuuid>"},"error_code":{"default":"GENERAL_ERROR","description":"Machine-readable error category, such as `BUSY`. `NEEDS_REFITTING` and `KV_CACHE_INVALID` mean the fitted model has to be fit again.","title":"Error Code","type":"string","x-sdk-keep-default":true,"uniqueKey":"error_code","typeLabel":"string"},"trace_id":{"oneOf":[{"type":"string","title":"Trace Id","description":"Id to quote when reporting the error.","format":"uuid","uniqueKey":"trace_id.oneOf.0","typeLabel":"string\u003cuuid>"},{"type":"null","title":"Trace Id","description":"Id to quote when reporting the error.","uniqueKey":"trace_id.oneOf.1","typeLabel":"null"}],"uniqueKey":"trace_id","typeLabel":"string\u003cuuid> | null"}},"required":["message","fitted_test_set_id"],"title":"PredictFailedErrorResponse","type":"object","uniqueKey":"","typeLabel":"PredictFailedErrorResponse · object"},"examples":{"Example":{"title":"Example","value":{"message":"\u003cstring>","fitted_test_set_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a","error_code":"GENERAL_ERROR","trace_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a"}}},"exampleType":"generated"}},"exampleType":"generated"},"2172b909-eeed-4264-a126-543eaeaf7d5f":{"description":"Error before or after a prediction job was created. Job failures include fitted_test_set_id.","content":{"application/json":{"schema":{"oneOf":[{"type":"object","title":"PredictFailedErrorResponse","description":"A prediction that failed after the server created it. get_predict_result\nreports the same failure under its id.","required":["message","fitted_test_set_id"],"properties":{"message":{"description":"What went wrong.","title":"Message","type":"string","isRequired":true,"uniqueKey":"oneOf.0.message","typeLabel":"string"},"fitted_test_set_id":{"description":"Id of the prediction that failed.","format":"uuid","title":"Fitted Test Set Id","type":"string","isRequired":true,"uniqueKey":"oneOf.0.fitted_test_set_id","typeLabel":"string\u003cuuid>"},"error_code":{"default":"GENERAL_ERROR","description":"Machine-readable error category, such as `BUSY`. `NEEDS_REFITTING` and `KV_CACHE_INVALID` mean the fitted model has to be fit again.","title":"Error Code","type":"string","x-sdk-keep-default":true,"uniqueKey":"oneOf.0.error_code","typeLabel":"string"},"trace_id":{"oneOf":[{"type":"string","title":"Trace Id","description":"Id to quote when reporting the error.","format":"uuid","uniqueKey":"oneOf.0.trace_id.oneOf.0","typeLabel":"string\u003cuuid>"},{"type":"null","title":"Trace Id","description":"Id to quote when reporting the error.","uniqueKey":"oneOf.0.trace_id.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.0.trace_id","typeLabel":"string\u003cuuid> | null"}},"additionalProperties":false,"uniqueKey":"oneOf.0","typeLabel":"PredictFailedErrorResponse · object"},{"type":"object","title":"ErrorResponse","required":["message"],"properties":{"message":{"description":"What went wrong.","title":"Message","type":"string","isRequired":true,"uniqueKey":"oneOf.1.message","typeLabel":"string"},"error_code":{"default":"GENERAL_ERROR","description":"Machine-readable error category, such as `BUSY`.","title":"Error Code","type":"string","x-sdk-keep-default":true,"uniqueKey":"oneOf.1.error_code","typeLabel":"string"},"trace_id":{"oneOf":[{"type":"string","title":"Trace Id","description":"Id to quote when reporting the error.","format":"uuid","uniqueKey":"oneOf.1.trace_id.oneOf.0","typeLabel":"string\u003cuuid>"},{"type":"null","title":"Trace Id","description":"Id to quote when reporting the error.","uniqueKey":"oneOf.1.trace_id.oneOf.1","typeLabel":"null"}],"uniqueKey":"oneOf.1.trace_id","typeLabel":"string\u003cuuid> | null"}},"additionalProperties":false,"uniqueKey":"oneOf.1","typeLabel":"ErrorResponse · object"}],"uniqueKey":"","typeLabel":"PredictFailedErrorResponse · object | ErrorResponse · object"},"examples":{"Example":{"title":"Example","value":{"message":"\u003cstring>","fitted_test_set_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a","error_code":"GENERAL_ERROR","trace_id":"3c90c3cc-0d44-4b50-8888-8dd25736052a"}}},"exampleType":"generated"}},"exampleType":"generated"}},"security":{"dbc4483c-2d19-44ca-8290-c974a9870b1b":{"BearerAuth":{"type":"http","scheme":"bearer","bearerFormat":"JWT","description":"Bearer token for authentication, obtained after signing up and generating an API key.","scopes":[],"schemeName":"BearerAuth"}}},"servers":{"13114613-4d51-455e-9bff-41c0cd6fa421":{"url":"https://api.priorlabs.ai","description":"Production TabPFN API"}},"processedSecurityOptions":[{"schemes":[{"scheme":{"type":"http","scheme":"bearer","bearerFormat":"JWT","description":"Bearer token for authentication, obtained after signing up and generating an API key.","scopes":[],"schemeName":"BearerAuth"},"schema":{"title":"Bearer Auth","type":"string","isRequired":true,"description":"Bearer token for authentication, obtained after signing up and generating an API key.","format":"bearer","typeLabel":"string\u003cbearer>","deprecated":false},"schemeName":"BearerAuth","name":"Authorization"}],"title":"BearerAuth"}]},"originalFileLocation":"api-reference/openapi.json"},"subdomain":"priorlabs","siteName":"Prior Labs","title":"Predict (TabPFN JSON API) - Prior Labs","navMaps":{"firstHrefInVersion":{},"firstHrefInLanguage":{}},"lastModified":"2026-11-04T13:18:55.034Z","originalFileLocation":"api-reference/prediction/predict-tabpfn-json-api.mdx"},"headHtml":"\u003cscript data-mint-head>(function(){var d=document.documentElement,p=\"light\";if(!p){try{p=localStorage.getItem(\"isDarkMode\")}catch(e){}if(p==='true')p='dark';else if(p==='false')p='light';if(p!=='light'\u0026\u0026p!=='dark'\u0026\u0026p!=='system')p=\"light\"}var r=p==='system'?(matchMedia(\"(prefers-color-scheme: dark)\").matches?'dark':'light'):p;d.classList.remove(r==='dark'?'light':'dark');d.classList.add(r);d.setAttribute(\"data-theme-preference\",p)})();\u003c/script>\u003cmeta data-mint-head name=\"description\" content=\"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\u0026quot;mean\u0026quot;, without quantiles.\">\u003cmeta data-mint-head name=\"application-name\" content=\"Prior Labs\">\u003cmeta data-mint-head name=\"generator\" content=\"Mintlify\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"192x192\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/android-chrome-192x192.png\" media=\"(prefers-color-scheme: light)\">\u003clink data-mint-head rel=\"apple-touch-icon\" type=\"image/png\" sizes=\"180x180\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/apple-touch-icon.png\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"16x16\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/favicon-16x16.png\" media=\"(prefers-color-scheme: light)\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"32x32\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/favicon-32x32.png\" media=\"(prefers-color-scheme: light)\">\u003clink data-mint-head rel=\"shortcut icon\" type=\"image/x-icon\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/favicon.ico\" media=\"(prefers-color-scheme: light)\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"192x192\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon-dark/android-chrome-192x192.png\" media=\"(prefers-color-scheme: dark)\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"16x16\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon-dark/favicon-16x16.png\" media=\"(prefers-color-scheme: dark)\">\u003clink data-mint-head rel=\"icon\" type=\"image/png\" sizes=\"32x32\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon-dark/favicon-32x32.png\" media=\"(prefers-color-scheme: dark)\">\u003clink data-mint-head rel=\"shortcut icon\" type=\"image/x-icon\" href=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon-dark/favicon.ico\" media=\"(prefers-color-scheme: dark)\">\u003cmeta data-mint-head property=\"og:title\" content=\"Predict (TabPFN JSON API) - Prior Labs\">\u003cmeta data-mint-head property=\"og:description\" content=\"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\u0026quot;mean\u0026quot;, without quantiles.\">\u003cmeta data-mint-head property=\"og:site_name\" content=\"Prior Labs\">\u003cmeta data-mint-head property=\"og:url\" content=\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api\">\u003cmeta data-mint-head property=\"og:type\" content=\"website\">\u003cmeta data-mint-head property=\"og:image\" content=\"https://priorlabs.mintlify.site/_mintlify/api/og/v2?division=REST+API\u0026amp;title=Predict+%28TabPFN+JSON+API%29\u0026amp;description=Predict+with+a+fitted+training+set.+Supply+test_set_upload_id%2C+or+x_test_parquet_base64+for+a+small+inline+test+set+within+the+limit+returned+by+%2Ftabpfn%2Fget_set\u0026amp;theme=9a7712e11d44b29294a56d4b\u0026amp;signature=180d260221f2782a19fec8e8198a72b05a11a506b5293524b5e117ca722ab8ab\">\u003cmeta data-mint-head property=\"og:image:width\" content=\"1200\">\u003cmeta data-mint-head property=\"og:image:height\" content=\"630\">\u003cmeta data-mint-head name=\"twitter:card\" content=\"summary_large_image\">\u003cmeta data-mint-head name=\"twitter:title\" content=\"Predict (TabPFN JSON API) - Prior Labs\">\u003cmeta data-mint-head name=\"twitter:description\" content=\"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\u0026quot;mean\u0026quot;, without quantiles.\">\u003cmeta data-mint-head name=\"twitter:image\" content=\"https://priorlabs.mintlify.site/_mintlify/api/og/v2?division=REST+API\u0026amp;title=Predict+%28TabPFN+JSON+API%29\u0026amp;description=Predict+with+a+fitted+training+set.+Supply+test_set_upload_id%2C+or+x_test_parquet_base64+for+a+small+inline+test+set+within+the+limit+returned+by+%2Ftabpfn%2Fget_set\u0026amp;theme=9a7712e11d44b29294a56d4b\u0026amp;signature=180d260221f2782a19fec8e8198a72b05a11a506b5293524b5e117ca722ab8ab\">\u003cmeta data-mint-head name=\"twitter:image:width\" content=\"1200\">\u003cmeta data-mint-head name=\"twitter:image:height\" content=\"630\">\u003cmeta data-mint-head name=\"msapplication-config\" content=\"/mintlify-assets/_mintlify/favicons/priorlabs/ezqAkuYrXl_TJelR/_generated/favicon/browserconfig.xml\">\u003cmeta data-mint-head name=\"apple-mobile-web-app-title\" content=\"Prior Labs\">\u003cmeta data-mint-head name=\"msapplication-TileColor\" content=\"#101075\">\u003clink data-mint-head rel=\"canonical\" href=\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api\">\u003clink data-mint-head rel=\"alternate\" type=\"application/xml\" href=\"/sitemap.xml\">\u003clink data-mint-head rel=\"alternate\" type=\"text/markdown\" href=\"/api-reference/prediction/predict-tabpfn-json-api.md\">\u003cscript data-mint-head type=\"application/ld+json\">{\"@context\":\"https://schema.org\",\"@graph\":[{\"@type\":\"Organization\",\"@id\":\"https://docs.priorlabs.ai/#organization\",\"name\":\"Prior Labs\",\"url\":\"https://docs.priorlabs.ai\",\"logo\":{\"@type\":\"ImageObject\",\"url\":\"https://mintcdn.com/priorlabs/t197PY0EpQCr-YHY/logo/light.svg?fit=max\u0026auto=format\u0026n=t197PY0EpQCr-YHY\u0026q=85\u0026s=33643e275cfc08c99769e1719c4f53c2\"}},{\"@type\":\"WebSite\",\"@id\":\"https://docs.priorlabs.ai#website\",\"name\":\"Prior Labs\",\"url\":\"https://docs.priorlabs.ai\",\"publisher\":{\"@id\":\"https://docs.priorlabs.ai/#organization\"}},{\"@type\":\"WebPage\",\"@id\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api#webpage\",\"url\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api\",\"name\":\"Predict (TabPFN JSON API)\",\"description\":\"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\\\"mean\\\", without quantiles.\",\"dateModified\":\"2026-11-04T13:18:55.034Z\",\"isPartOf\":{\"@id\":\"https://docs.priorlabs.ai#website\"},\"breadcrumb\":{\"@id\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api#breadcrumb\"}},{\"@type\":\"BreadcrumbList\",\"@id\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"REST API\",\"item\":\"https://docs.priorlabs.ai/api-reference/training/prepare-train-set-upload\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"Predict (TabPFN JSON API)\",\"item\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api\"}]},{\"@type\":[\"TechArticle\",\"APIReference\"],\"@id\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api#apireference\",\"headline\":\"Predict (TabPFN JSON API)\",\"name\":\"Predict (TabPFN JSON API)\",\"description\":\"Predict with a fitted training set. Supply test_set_upload_id, or x_test_parquet_base64 for a small inline test set within the limit returned by /tabpfn/get_settings. Choose classification or regression in task_config. A completed response contains predictions, metadata and optional timings; a pending response contains fitted_test_set_id and retry_in_secs for polling /tabpfn/get_predict_result. Long requests may stream keepalive whitespace before the final JSON. Errors after streaming starts appear in the response body even when the HTTP status is 200. Thinking regression supports only output_type=\\\"mean\\\", without quantiles.\",\"url\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api\",\"mainEntityOfPage\":{\"@id\":\"https://docs.priorlabs.ai/api-reference/prediction/predict-tabpfn-json-api#webpage\"},\"dateModified\":\"2026-11-04T13:18:55.034Z\",\"publisher\":{\"@id\":\"https://docs.priorlabs.ai/#organization\"},\"isPartOf\":{\"@id\":\"https://docs.priorlabs.ai#website\"}}]}\u003c/script>\u003cscript data-mint-head type=\"application/ld+json\">{\"@context\":\"https://schema.org\",\"@type\":\"WebSite\",\"name\":\"Prior Labs\",\"creator\":{\"@type\":\"Organization\",\"name\":\"Mintlify\",\"url\":\"https://mintlify.com\"}}\u003c/script>","siteName":"Prior Labs","nav":{"global":{"anchors":[{"anchor":"GitHub","icon":"github","href":"https://github.com/priorlabs/tabpfn"},{"anchor":"Discord","icon":"discord","href":"https://discord.com/invite/VJRuU3bSxt"},{"anchor":"Colab Demo","icon":"book","href":"https://colab.research.google.com/github/PriorLabs/TabPFN/blob/main/examples/notebooks/TabPFN_Demo_Local.ipynb"},{"anchor":"Changelog","icon":"clock-rotate-left","href":"/changelog"}]},"tabs":[{"tab":"Documentation","icon":"book-open","pages":[{"title":"Documentation","href":"/overview"}]},{"tab":"Cookbook","icon":"hat-chef","pages":[{"title":"Cookbook","href":"/cookbook"}]},{"tab":"API reference","icon":"code","groups":[{"group":"Reference","pages":[{"title":"API reference","href":"/api-reference/overview"}]},{"group":"tabpfn","pages":[{"sidebarTitle":"Overview","title":"tabpfn","href":"/api-reference/python/tabpfn/overview"},{"group":"TabPFNClassifier","expanded":false,"pages":[{"title":"Constructor and configuration","href":"/api-reference/python/tabpfn/classifier/configuration"},{"title":"Fitting and prediction","href":"/api-reference/python/tabpfn/classifier/fitting-and-prediction"},{"title":"Embeddings","href":"/api-reference/python/tabpfn/classifier/embeddings"},{"title":"Batched and differentiable inference","href":"/api-reference/python/tabpfn/classifier/advanced"},{"title":"Persistence and devices","href":"/api-reference/python/tabpfn/classifier/persistence"}]},{"group":"TabPFNRegressor","expanded":false,"pages":[{"title":"Constructor and configuration","href":"/api-reference/python/tabpfn/regressor/configuration"},{"title":"Fitting and prediction","href":"/api-reference/python/tabpfn/regressor/fitting-and-prediction"},{"title":"Embeddings","href":"/api-reference/python/tabpfn/regressor/embeddings"},{"title":"Batched and differentiable inference","href":"/api-reference/python/tabpfn/regressor/advanced"},{"title":"Persistence and devices","href":"/api-reference/python/tabpfn/regressor/persistence"},{"title":"Predictive distributions","href":"/api-reference/python/tabpfn/regressor/predictive-distributions"}]},{"title":"Inference configuration","href":"/api-reference/python/tabpfn/inference-config"},{"title":"Inference tuning","href":"/api-reference/python/tabpfn/inference-tuning"},{"title":"Model versions","href":"/api-reference/python/tabpfn/model-versions"},{"title":"Model persistence","href":"/api-reference/python/tabpfn/model-persistence"},{"title":"Diagnostics and errors","href":"/api-reference/python/tabpfn/diagnostics"},{"group":"Fine-tuning","expanded":false,"pages":[{"title":"Batch and evaluation types","href":"/api-reference/python/tabpfn/finetuning/data"},{"title":"Fine-tuning base class","href":"/api-reference/python/tabpfn/finetuning/base"},{"title":"Fine-tuned classifier","href":"/api-reference/python/tabpfn/finetuning/classifier"},{"title":"Fine-tuned regressor","href":"/api-reference/python/tabpfn/finetuning/regressor"},{"title":"Experiment logging","href":"/api-reference/python/tabpfn/finetuning/logging"}]},{"group":"Preprocessing","expanded":false,"pages":[{"title":"Preprocessing configuration","href":"/api-reference/python/tabpfn/preprocessing/configuration"},{"title":"Preprocessing pipelines","href":"/api-reference/python/tabpfn/preprocessing/pipeline"},{"title":"Preprocessor presets","href":"/api-reference/python/tabpfn/preprocessing/presets"}]}]},{"group":"tabpfn-client","pages":[{"sidebarTitle":"Overview","title":"tabpfn-client","href":"/api-reference/python/tabpfn-client/overview"},{"group":"TabPFNClassifier","expanded":false,"pages":[{"title":"Constructor and configuration","href":"/api-reference/python/tabpfn-client/classifier/configuration"},{"title":"Fitting and prediction","href":"/api-reference/python/tabpfn-client/classifier/fitting-and-prediction"},{"title":"Model selection and persistence","href":"/api-reference/python/tabpfn-client/classifier/model-selection-and-persistence"}]},{"group":"TabPFNRegressor","expanded":false,"pages":[{"title":"Constructor and configuration","href":"/api-reference/python/tabpfn-client/regressor/configuration"},{"title":"Fitting and prediction","href":"/api-reference/python/tabpfn-client/regressor/fitting-and-prediction"},{"title":"Model selection and persistence","href":"/api-reference/python/tabpfn-client/regressor/model-selection-and-persistence"},{"title":"Predictive distributions","href":"/api-reference/python/tabpfn-client/regressor/predictive-distributions"}]},{"title":"Authentication","href":"/api-reference/python/tabpfn-client/authentication"},{"title":"Configuration","href":"/api-reference/python/tabpfn-client/configuration"},{"title":"Usage and costs","href":"/api-reference/python/tabpfn-client/usage-and-costs"},{"title":"UserDataClient","href":"/api-reference/python/tabpfn-client/user-data"},{"group":"Deployment backends","expanded":false,"pages":[{"title":"Self-hosted HTTP","href":"/api-reference/python/tabpfn-client/backends/hosted"},{"title":"SageMaker","href":"/api-reference/python/tabpfn-client/backends/sagemaker"},{"title":"Azure Foundry","href":"/api-reference/python/tabpfn-client/backends/foundry"}]},{"title":"Exceptions","href":"/api-reference/python/tabpfn-client/exceptions"},{"title":"SHAP explanations","href":"/api-reference/python/tabpfn-client/explain"}]},{"group":"tabpfn-extensions","pages":[{"sidebarTitle":"Overview","title":"tabpfn-extensions","href":"/api-reference/python/tabpfn-extensions/overview"},{"group":"Interpretability","root":{"title":"Interpretability","href":"/api-reference/python/tabpfn-extensions/interpretability"},"expanded":false,"pages":[{"title":"Shapley explainers","href":"/api-reference/python/tabpfn-extensions/interpretability/shapley-explainers"},{"title":"Partial dependence","href":"/api-reference/python/tabpfn-extensions/interpretability/partial-dependence"},{"title":"Feature selection","href":"/api-reference/python/tabpfn-extensions/interpretability/feature-selection"},{"title":"Decoder readout","href":"/api-reference/python/tabpfn-extensions/interpretability/decoder-readout"}]},{"title":"Conformal prediction with missing data","href":"/api-reference/python/tabpfn-extensions/cp-missing-data"},{"title":"Hurdle regression","href":"/api-reference/python/tabpfn-extensions/hurdle"},{"title":"Many-class classification","href":"/api-reference/python/tabpfn-extensions/many-class"},{"title":"Embeddings","href":"/api-reference/python/tabpfn-extensions/embedding"},{"group":"Images","root":{"title":"Images","href":"/api-reference/python/tabpfn-extensions/image"},"expanded":false,"pages":[{"title":"Image transformer","href":"/api-reference/python/tabpfn-extensions/image/transformer"},{"title":"Image estimator","href":"/api-reference/python/tabpfn-extensions/image/estimator"}]},{"group":"Unsupervised learning","root":{"title":"Unsupervised learning","href":"/api-reference/python/tabpfn-extensions/unsupervised"},"expanded":false,"pages":[{"title":"Model and fitting","href":"/api-reference/python/tabpfn-extensions/unsupervised/model"},{"title":"Imputation and data generation","href":"/api-reference/python/tabpfn-extensions/unsupervised/imputation"},{"title":"Anomaly detection","href":"/api-reference/python/tabpfn-extensions/unsupervised/outliers"},{"title":"Unsupervised embeddings","href":"/api-reference/python/tabpfn-extensions/unsupervised/embeddings"},{"title":"Unsupervised experiments","href":"/api-reference/python/tabpfn-extensions/unsupervised/experiments"}]},{"title":"Energy-based data generation","href":"/api-reference/python/tabpfn-extensions/tabebm"},{"title":"Bayesian optimization","href":"/api-reference/python/tabpfn-extensions/bayesian-optimization"},{"title":"Conditional randomization tests","href":"/api-reference/python/tabpfn-extensions/pval-crt"},{"title":"Survival analysis","href":"/api-reference/python/tabpfn-extensions/survival"},{"title":"Experiments","href":"/api-reference/python/tabpfn-extensions/benchmarking"},{"title":"Scoring","href":"/api-reference/python/tabpfn-extensions/scoring"},{"title":"Utilities","href":"/api-reference/python/tabpfn-extensions/utils"}]},{"group":"tabpfn-rel","expanded":false,"pages":[{"sidebarTitle":"Overview","title":"tabpfn-rel","href":"/api-reference/python/tabpfn-rel/overview"},{"title":"TabPFNRel","href":"/api-reference/python/tabpfn-rel/estimator"},{"title":"Model backends","href":"/api-reference/python/tabpfn-rel/models"},{"title":"Search spaces","href":"/api-reference/python/tabpfn-rel/search-spaces"},{"group":"Context and queries (RPI)","expanded":false,"pages":[{"title":"Predictive context","href":"/api-reference/python/relarena-core/context"},{"title":"Prediction queries","href":"/api-reference/python/relarena-core/query"},{"title":"Fitted predictor","href":"/api-reference/python/relarena-core/fitted-predictor"},{"title":"Database and task specifications","href":"/api-reference/python/relarena-core/specifications"}]}]},{"group":"REST API","icon":"code","pages":[{"title":"Prepare train set upload","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/prepare_train_set_upload","href":"/api-reference/training/prepare-train-set-upload"},{"title":"Get model limits","deprecated":true,"version":null,"openapi":"/api-reference/openapi.json get /tabpfn/get_model_limits","href":"/api-reference/training/get-model-limits"},{"title":"Fit (TabPFN JSON API)","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/fit","href":"/api-reference/training/fit-tabpfn-json-api"},{"title":"Prepare test set upload","description":"Request signed upload URLs for test features associated with a fitted training set. PUT the file bytes to the returned signed URLs using `required_headers`.","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/prepare_test_set_upload","href":"/api-reference/prediction/prepare-test-set-upload"},{"title":"Predict (TabPFN JSON API)","deprecated":null,"version":null,"openapi":"/api-reference/openapi.json post /tabpfn/predict","href":"/api-reference/prediction/predict-tabpfn-json-api"}]}]},{"tab":"Blog","icon":"newspaper","href":"https://priorlabs.ai/blog#articles"}]},"basePath":"","title":"Predict (TabPFN JSON API) - Prior Labs","buildId":"dpl_DTau1J5HqS9UTgVSDwGksstFWtPa","deploymentId":"dpl_DTau1J5HqS9UTgVSDwGksstFWtPa","dynamics":["10e4515597","c2aa737661","2b790eda92","b0f84f53ff","a9bb46feb3"],"chunks":["BannerMarkdown-Cj6eAYhz.js","dist-CdrJecqK.js","lib-DuNMoZ05.js","lib-B8HbPgcC.js","clsx-D-TRO5q5.js","html-void-elements-mCSPHezU.js","katex-YMM-FMeD.js","marked.esm-CZWythHk.js","OpenApiReferenceProvider-5FCzAcfS.js","utils-SLtAG4EZ.js","cloneDeep-izs3Hzxb.js","_Uint8Array-BVIqJ90l.js","isObjectLike-C8vmzPEN.js","isArray-Dcu1dQ5t.js","utils-2DZi8ZMu.js","toString-BX-iyh8G.js","isSymbol-CSH9U2Gm.js","addUniqueKeysToSchema-D7J3nkzE.js","snippetPresets-D0Alv0AH.js","style-to-object-BY3JEAdk.js","v4-DIgz8zy6.js","conditional-requirements-CFM0id0Q.js","ApiExamples-BI4qNJ2c.js","isEqual-BfaiBbcl.js","usePlaygroundInputsStore-BD3uyr8j.js","shallow-D8iEBr5z.js","getApiPlaygroundDisplay-CgNP4lzR.js","CodeSnippet-CX-QHuMs.js","CodeSnippets-VlXE-ZUf.js","Page-D9ithiIZ.js","getPastedJson-DYNQkZ1M.js","icon-CzacdwOi.js","MethodPill-2JVqoYKi.js","DuplicateIcon-_xVXs-U8.js","createIcon-BRHrweLo.js","DynamicMarkdown-mQG_iQiE.js","SectionHeading-DMpJzv8f.js","axios-vUo_bRtp.js","Markdown-5FGIym-8.js","AssistantPageLink-CHlcWBLM.js"]}