|
|
|
@@ -728,7 +728,7 @@ class Dataset: |
|
|
|
sizes (Union[list[int], list[float]]): If a list of integers [s1, s2, …, sn] is |
|
|
|
provided, the dataset will be split into n datasets of size s1, size s2, …, size sn |
|
|
|
respectively. If the sum of all sizes does not equal the original dataset size, an |
|
|
|
an error will occur. |
|
|
|
error will occur. |
|
|
|
If a list of floats [f1, f2, …, fn] is provided, all floats must be between 0 and 1 |
|
|
|
and must sum to 1, otherwise an error will occur. The dataset will be split into n |
|
|
|
Datasets of size round(f1*K), round(f2*K), …, round(fn*K) where K is the size of the |
|
|
|
|