Skip to content
Snippets Groups Projects

Add support for fitting estimators on dask bags

Merged Amir MOHAMMADI requested to merge fit-dask-bags into master

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

Loading
Loading

Activity

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