Skip to content

Commit

Permalink
Improve docs
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-narozniak committed May 20, 2024
1 parent 74e70aa commit 0f9ffc0
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions datasets/flwr_datasets/federated_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,8 +46,10 @@ class FederatedDataset:
Secondary information regarding the dataset, most often subset or version
(that is passed to the name in datasets.load_dataset).
preprocessor : Optional[Union[Preprocessor, Dict[str, Tuple[str, ...]]]]
`Callable` that transforms `DatasetDict` splits, or configuration dict for
`MergeResplitter`.
`Callable` that transforms `DatasetDict` by resplitting, removing
features, creating new features, performing any other preprocessing operation,
or configuration dict for `MergeResplitter`. Applied after shuffling. If None,
no operation is applied.
partitioners : Dict[str, Union[Partitioner, int]]
A dictionary mapping the Dataset split (a `str`) to a `Partitioner` or an `int`
(representing the number of IID partitions that this split should be partitioned
Expand Down

0 comments on commit 0f9ffc0

Please sign in to comment.