| 775 | } |
| 776 | |
| 777 | CATBOOST_API bool PredictSpecificClassText( |
| 778 | ModelCalcerHandle* modelHandle, |
| 779 | size_t docCount, |
| 780 | const float** floatFeatures, size_t floatFeaturesSize, |
| 781 | const char*** catFeatures, size_t catFeaturesSize, |
| 782 | const char*** textFeatures, size_t textFeaturesSize, |
| 783 | int classId, |
| 784 | double* result, size_t resultSize) { |
| 785 | try { |
| 786 | const size_t dim = FULL_MODEL_PTR(modelHandle)->GetDimensionsCount(); |
| 787 | TVector<double> rawResult(docCount * dim); |
| 788 | if (!CalcModelPredictionText( |
| 789 | modelHandle, docCount, |
| 790 | floatFeatures, floatFeaturesSize, |
| 791 | catFeatures, catFeaturesSize, |
| 792 | textFeatures, textFeaturesSize, |
| 793 | rawResult.data(), rawResult.size())) |
| 794 | { |
| 795 | return false; |
| 796 | } |
| 797 | GetSpecificClass(classId, rawResult, dim, TArrayRef<double>(result, resultSize)); |
| 798 | } catch (...) { |
| 799 | ErrorMessageHolder.Get().Message = CurrentExceptionMessage(); |
| 800 | return false; |
| 801 | } |
| 802 | return true; |
| 803 | } |
| 804 | |
| 805 | CATBOOST_API bool PredictSpecificClassTextAndEmbeddings( |
| 806 | ModelCalcerHandle* modelHandle, |
nothing calls this directly
no test coverage detected