Skip to content

expense_estimation

estimate_duration

estimate_duration(
    num_rows: int,
    num_features: int,
    task: Literal["classification", "regression"],
    tabpfn_config: dict = {},
    duration_factor: float = VERTEX_GPU_FACTOR,
    latency_offset: float = 0.0,
) -> float

Estimates the duration of a prediction task.