| 192 | |
| 193 | namespace { |
| 194 | void GetSpecificClass(int classId, TArrayRef<double> predictions, size_t dim, TArrayRef<double> result) { |
| 195 | for (size_t docId = 0; docId < result.size(); ++docId) { |
| 196 | result[docId] = predictions[docId * dim + classId]; |
| 197 | } |
| 198 | } |
| 199 | |
| 200 | bool GetFeatureIndices(std::function<TVector<size_t>()> getFeatureIndices, size_t** indices, size_t* count) { |
| 201 | try { |
no test coverage detected