| 560 | } |
| 561 | |
| 562 | static TIntrusivePtr<TTrainingDataProvider> MakeFeatureSubsetDataProvider( |
| 563 | const TVector<ui32>& ignoredFeatures, |
| 564 | NCB::TTrainingDataProviderPtr trainingDataProvider |
| 565 | ) { |
| 566 | TQuantizedObjectsDataProviderPtr newObjects = dynamic_cast<TQuantizedObjectsDataProvider*>( |
| 567 | trainingDataProvider->ObjectsData->GetFeaturesSubset(ignoredFeatures, &NPar::LocalExecutor()).Get()); |
| 568 | CB_ENSURE( |
| 569 | newObjects, |
| 570 | "Objects data provider must be TQuantizedObjectsDataProvider or TQuantizedObjectsDataProvider"); |
| 571 | TDataMetaInfo newMetaInfo = trainingDataProvider->MetaInfo; |
| 572 | newMetaInfo.FeaturesLayout = newObjects->GetFeaturesLayout(); |
| 573 | return MakeIntrusive<TTrainingDataProvider>( |
| 574 | trainingDataProvider->OriginalFeaturesLayout, |
| 575 | TDataMetaInfo(newMetaInfo), |
| 576 | trainingDataProvider->ObjectsGrouping, |
| 577 | newObjects, |
| 578 | trainingDataProvider->TargetData); |
| 579 | } |
| 580 | |
| 581 | TTrainingDataProviders MakeFeatureSubsetTrainingData( |
| 582 | const TVector<ui32>& ignoredFeatures, |
no test coverage detected