| 105 | |
| 106 | |
| 107 | static void CheckBaseline( |
| 108 | const TVector<TSharedVector<float>>& baseline, |
| 109 | ui32 objectCount, |
| 110 | TMaybe<ui32> classCount |
| 111 | ) { |
| 112 | if (baseline.size() == 1) { |
| 113 | CB_ENSURE_INTERNAL( |
| 114 | !classCount || (*classCount == 2), |
| 115 | "One-dimensional baseline with multiple classes" |
| 116 | ); |
| 117 | } else { |
| 118 | CB_ENSURE_INTERNAL(classCount, "Multidimensional baseline for non-multiclassification"); |
| 119 | CheckDataSize(baseline.size(), (size_t)*classCount, "Baseline", true, "class count"); |
| 120 | } |
| 121 | |
| 122 | for (auto i : xrange(baseline.size())) { |
| 123 | CheckOneBaseline(*(baseline[i]), i, objectCount); |
| 124 | } |
| 125 | } |
| 126 | |
| 127 | // groupWeights is indexed by objectIdx |
| 128 | void NCB::CheckGroupWeights( |
no test coverage detected