diff --git a/tabpfn_client/estimator.py b/tabpfn_client/estimator.py index 5c2aed6..d3227a3 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. @@ -262,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.