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

Function IsMultiClassification

catboost/libs/fstr/shap_prepared_trees.cpp:492–504  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

490
491
492static inline bool IsMultiClassification(const TFullModel& model) {
493 ELossFunction modelLoss = ELossFunction::RMSE;
494 if (IsMultiClass(model)) {
495 TMaybe<ELossFunction> loss = TryGuessModelMultiClassLoss(model);
496 if (loss) {
497 modelLoss = *loss.Get();
498 } else {
499 CATBOOST_WARNING_LOG << "There is no loss_function parameter in the model, so it is considered as MultiClass" << Endl;
500 modelLoss = ELossFunction::MultiClass;
501 }
502 }
503 return (modelLoss == ELossFunction::MultiClass);
504}
505
506
507TShapPreparedTrees PrepareTreesWithoutIndependent(

Callers 2

PrepareTreesFunction · 0.85

Calls 3

IsMultiClassFunction · 0.85
GetMethod · 0.45

Tested by

no test coverage detected