mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
7ce22b72ed
Expands the `to_torch` method for Datasets with: * An ability to choose to output a list/dict of feature tensors instead of just one (through setting `feature_columns` to be a list of lists or a dict of lists) * An ability to choose whether the label should be unsqueezed or not * An ability to pass `None` as the label (for prediction). Furthermore, this changes how the `feature_column_dtypes` argument works. Previously, it took a list of dtypes for each feature. However, as the tensor was concatenated in the end, only one dtype mattered (the biggest one). Now, this argument expects a single dtype which will be applied to the features tensor (or a list/dict if `feature_columns` is a list of list/dict of lists). Unit tests for all cases are included. Co-authored-by: matthewdeng <matthew.j.deng@gmail.com> |
||
|---|---|---|
| .. | ||
| chaos_test | ||
| dask_on_ray | ||
| dataset | ||
| decision_tree | ||
| many_nodes_tests | ||
| placement_group_tests | ||
| shuffle | ||
| shuffle_data_loader | ||
| stress_tests | ||
| chaos_test.yaml | ||
| nightly_tests.yaml | ||
| setup_chaos.py | ||
| wait_cluster.py | ||