| 210 | } |
| 211 | |
| 212 | void SetGroupWeights( |
| 213 | const TPathWithScheme& groupWeightsPath, |
| 214 | ui32 objectCount, |
| 215 | TDatasetSubset loadSubset, |
| 216 | IDatasetVisitor* visitor |
| 217 | ) { |
| 218 | DumpMemUsage("After data read"); |
| 219 | if (groupWeightsPath.Inited()) { |
| 220 | auto maybeGroupIds = visitor->GetGroupIds(); |
| 221 | CB_ENSURE(maybeGroupIds, "Cannot load group weights data for a dataset without groups"); |
| 222 | TVector<float> groupWeights = ReadGroupWeights( |
| 223 | groupWeightsPath, |
| 224 | *maybeGroupIds, |
| 225 | objectCount, |
| 226 | loadSubset |
| 227 | ); |
| 228 | visitor->SetGroupWeights(std::move(groupWeights)); |
| 229 | } |
| 230 | } |
| 231 | |
| 232 | void SetBaseline( |
| 233 | const TPathWithScheme& baselinePath, |
no test coverage detected