From c17832767b5994723a321057a0f995c903d77eae Mon Sep 17 00:00:00 2001 From: LeoGrin Date: Tue, 21 Jan 2025 19:10:22 +0100 Subject: [PATCH 1/2] improve doc for inference config --- tabpfn_client/estimator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tabpfn_client/estimator.py b/tabpfn_client/estimator.py index 5c2aed6..3d1b84c 100644 --- a/tabpfn_client/estimator.py +++ b/tabpfn_client/estimator.py @@ -110,7 +110,9 @@ def __init__( random_state: int or RandomState or RandomGenerator or None, default=None Controls the randomness of the model. Pass an int for reproducible results. inference_config: dict or None, default=None - Additional advanced arguments for model interface. + Additional advanced arguments for model interface. See the doc of ModelInterfaceConfig + in the tabpfn package for more details. For the client, the inference_config and the + preprocess transforms need to be dictionaries. paper_version: bool, default=False If True, will use the model described in the paper, instead of the newest version available on the API, which e.g handles text features better. From c183970528d0607867e4a91920f658d9ff61290f Mon Sep 17 00:00:00 2001 From: LeoGrin Date: Tue, 21 Jan 2025 19:12:17 +0100 Subject: [PATCH 2/2] also for regressor --- tabpfn_client/estimator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tabpfn_client/estimator.py b/tabpfn_client/estimator.py index 3d1b84c..d3227a3 100644 --- a/tabpfn_client/estimator.py +++ b/tabpfn_client/estimator.py @@ -264,7 +264,9 @@ def __init__( random_state: int or RandomState or RandomGenerator or None, default=None Controls the randomness of the model. Pass an int for reproducible results. inference_config: dict or None, default=None - Additional advanced arguments for model interface. + Additional advanced arguments for model interface. See the doc of ModelInterfaceConfig + in the tabpfn package for more details. For the client, the inference_config and the + preprocess transforms need to be dictionaries. paper_version: bool, default=False If True, will use the model described in the paper, instead of the newest version available on the API, which e.g handles text features better.