| 98 | |
| 99 | |
| 100 | static void GetWorkerPart( |
| 101 | TTrainingDataProviderPtr srcData, |
| 102 | TArraySubsetIndexing<ui32>&& subsetIndexing, |
| 103 | TMaybe<TQuantizedObjectsDataProviderPtr> estimatedSrcData, |
| 104 | ui64 cpuUsedRamLimit, |
| 105 | NPar::ILocalExecutor* localExecutor, |
| 106 | TTrainingDataProviderPtr* dstData, |
| 107 | TMaybe<TQuantizedObjectsDataProviderPtr*> estimatedDstData |
| 108 | ) { |
| 109 | const TObjectsGroupingSubset objectsGroupingSubset = NCB::GetSubset( |
| 110 | srcData->ObjectsGrouping, |
| 111 | std::move(subsetIndexing), |
| 112 | EObjectsOrder::Ordered); |
| 113 | |
| 114 | *dstData = srcData->GetSubset(objectsGroupingSubset, cpuUsedRamLimit, localExecutor); |
| 115 | |
| 116 | if (estimatedSrcData) { |
| 117 | (*estimatedDstData)->Reset(dynamic_cast<TQuantizedObjectsDataProvider*>( |
| 118 | (*estimatedSrcData)->GetSubset(objectsGroupingSubset, cpuUsedRamLimit, localExecutor).Get())); |
| 119 | } |
| 120 | } |
| 121 | |
| 122 | |
| 123 | void SetTrainDataFromMaster( |