EstimatorReport.cache_predictions

EstimatorReport.cache_predictions(response_methods='auto', n_jobs=None)

Cache estimator’s predictions.

Parameters:

response_methods“auto” or list of str, default=”auto”

The response methods to precompute. If “auto”, the response methods are inferred from the ml task: for classification we compute the response of the predict_proba, decision_function and predict methods; for regression we compute the response of the predict method.

n_jobsint or None, default=None

The number of jobs to run in parallel. None means 1 unless in a joblib.parallel_backend context. -1 means using all processors.

Examples

>>> from sklearn.datasets import load_breast_cancer
>>> from sklearn.linear_model import LogisticRegression
>>> from sklearn.model_selection import train_test_split
>>> from skore import EstimatorReport
>>> X_train, X_test, y_train, y_test = train_test_split(
...     *load_breast_cancer(return_X_y=True), random_state=0
... )
>>> classifier = LogisticRegression(max_iter=10_000)
>>> report = EstimatorReport(
...     classifier,
...     X_train=X_train,
...     y_train=y_train,
...     X_test=X_test,
...     y_test=y_test,
... )
>>> report.cache_predictions()
>>> report._cache
{...}