MCPcopy Create free account
hub / github.com/catboost/catboost / DegroupDataset

Function DegroupDataset

catboost/libs/data/sampler.cpp:84–105  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

82}
83
84void DegroupDataset(TDataProvider* dataset) {
85 auto& metaInfo = dataset->MetaInfo;
86 metaInfo.HasGroupId = false;
87 metaInfo.HasGroupWeight = false;
88 metaInfo.HasSubgroupIds = false;
89
90 if (metaInfo.ColumnsInfo) {
91 for (auto& column : metaInfo.ColumnsInfo->Columns) {
92 switch (column.Type) {
93 case EColumn::GroupId:
94 case EColumn::SubgroupId:
95 case EColumn::GroupWeight:
96 column.Type = EColumn::Auxiliary;
97 default:
98 break;
99 }
100 }
101 }
102
103 auto objectCount = dataset->GetObjectCount();
104 dataset->ObjectsGrouping = MakeIntrusive<TObjectsGrouping>(objectCount);
105}
106
107
108TDataProviderPtr NCB::DataProviderSamplerReorderByIndices(

Calls 1

GetObjectCountMethod · 0.45

Tested by

no test coverage detected