pfn_phe ¶

AutoPostHocEnsemblePredictor ¶

Bases: BaseEstimator

A wrapper to effectively performing post hoc ensemble with TabPFN models.

fit(
    X: ndarray,
    y: ndarray,
    categorical_feature_indices: list[int] | None = None,
) -> AutoPostHocEnsemblePredictor

Fits the post hoc ensemble on the given data.

Parameters:

Name	Type	Description	Default
`X`	`ndarray`	The input data to fit the ensemble on.	required
`y`	`ndarray`	The target values to fit the ensemble on.	required
`categorical_feature_indices`	`list[int] \| None`	The indices of the categorical features in the data. If None, no categorical features are assumed to be present.	`None`

predict(X: ndarray) -> ndarray

Predicts the target values for the given data.

predict_proba(X: ndarray) -> ndarray

Predicts the target values for the given data.