(dataset, num)
| 111 | |
| 112 | |
| 113 | def cut_dataset(dataset, num): |
| 114 | |
| 115 | length = len(dataset) |
| 116 | |
| 117 | torch.manual_seed(0) |
| 118 | selected_dataset, _ = torch.utils.data.random_split( |
| 119 | dataset, [num, length - num]) |
| 120 | return selected_dataset |
| 121 | |
| 122 | |
| 123 | def count_dataset_for_class(num_class, dataset): |
no outgoing calls
no test coverage detected