Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improve doc for inference config #79

Merged
merged 2 commits into from
Jan 22, 2025
Merged

Improve doc for inference config #79

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
c178327
improve doc for inference config
LeoGrin Jan 21, 2025
c183970
also for regressor
LeoGrin Jan 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions tabpfn_client/estimator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down
Loading