> ## Documentation Index
> Fetch the complete documentation index at: https://docs.priorlabs.ai/llms.txt
> Use this file to discover all available pages before exploring further.

> ## Agent 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.

# Fitted predictor

> Fitted predictor for TabPFN-Rel.

> [!NOTE]
>
> Looking for usage documentation? Check out [Relational predictions](https://docs.priorlabs.ai/capabilities/relational.md).

## `FittedPredictor`

[ View source ](https://github.com/PriorLabs/relarena/blob/4b74c718ba6ff9c67cf63b4beac6dadaffdd09d8/packages/relarena-core/src/relarena_core/userdb/query.py#L346)

A fitted model and its tuning results for one training context.

```python theme={null}
FittedPredictor(
    context: PredictiveContext,
    _model: RelArenaModel,
    _cache: CacheConfig,
    trials: list[TrialResult] | None,
    config: dict[str, Any],
)
```

**Fields**

| Name | Type | Default | Description |
| - | - | - | - |
| `context` | `PredictiveContext` | Required | — |
| `trials` | `list[TrialResult] \| None` | Required | — |
| `config` | `dict[str, Any]` | Required | — |

***

## `FittedPredictor.predict`

[ View source ](https://github.com/PriorLabs/relarena/blob/4b74c718ba6ff9c67cf63b4beac6dadaffdd09d8/packages/relarena-core/src/relarena_core/userdb/query.py#L356)

Predict original entity IDs with features frozen at the context cutoff.

```python theme={null}
FittedPredictor.predict(
    query: PredictiveQuery,
    *,
    cache_dir: str | Path | None = None,
) -> pd.DataFrame
```

**Parameters**

| Parameter | Type | Default | Description |
| - | - | - | - |
| `query` | `PredictiveQuery` | Required | The entities and anchor timestamp to predict. |
| `cache_dir` | `str \| Path \| None` | `None` | Feature cache; `None` reuses the one passed to [`fit`](https://docs.priorlabs.ai/api-reference/python/relarena-core/context#predictivecontext-fit). |

**Returns**

| Type | Description |
| - | - |
| `pd.DataFrame` | One row per entity: the time and entity columns and `<target_col>_pred`. |


This documentation is built and hosted on [Mintlify](https://mintlify.com), a developer documentation platform.