Skip to content
Snippets Groups Projects

Add support for fitting estimators on dask bags

The estimators that can handle dask bags should set the bob_fit_supports_dask_bag as True. This commit also includes

  • Adds a new tag: bob_fit_supports_dask_bag
  • Adds a new tag: bob_checkpoint_features for when you want to always avoid checkpointing features for a specific estimator.
  • Expose dask_tags, get_bob_tags in the main API
  • The SampleWrapper was modified to support bob_fit_supports_dask_bag
  • The CheckpointWrapper now loads estimators without losing references correctly.

Merge request reports

Pipeline #61026 passed

Pipeline passed for d8c2cd53 on fit-dask-bags

Approval is optional

Merged by Amir MOHAMMADIAmir MOHAMMADI May 9, 2022 (May 9, 2022 2:31pm UTC)

Merge details

  • Changes merged into master with e4c71635.
  • Deleted the source branch.
  • Auto-merge enabled

Pipeline #61028 passed

Pipeline passed for e4c71635 on master

Deployed to py‎pi‎ Jun 15, 2022
Deployed to sta‎ble‎ Jun 15, 2022
Deployed to be‎ta‎ May 9, 2022

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading