| 1437 | } |
| 1438 | |
| 1439 | TVector<TVector<TVector<double>>> CalcShapValuesMulti( |
| 1440 | const TFullModel& model, |
| 1441 | const TDataProvider& dataset, |
| 1442 | const TDataProviderPtr referenceDataset, |
| 1443 | const TMaybe<TFixedFeatureParams>& fixedFeatureParams, |
| 1444 | int logPeriod, |
| 1445 | EPreCalcShapValues mode, |
| 1446 | NPar::ILocalExecutor* localExecutor, |
| 1447 | ECalcTypeShapValues calcType, |
| 1448 | EExplainableModelOutput modelOutputType |
| 1449 | ) { |
| 1450 | TShapPreparedTrees preparedTrees = PrepareTrees( |
| 1451 | model, |
| 1452 | &dataset, |
| 1453 | referenceDataset, |
| 1454 | mode, |
| 1455 | localExecutor, |
| 1456 | /*calcInternalValues*/ false, |
| 1457 | calcType, |
| 1458 | modelOutputType |
| 1459 | ); |
| 1460 | CalcShapValuesByLeaf( |
| 1461 | model, |
| 1462 | fixedFeatureParams, |
| 1463 | logPeriod, |
| 1464 | preparedTrees.CalcInternalValues, |
| 1465 | localExecutor, |
| 1466 | &preparedTrees, |
| 1467 | calcType |
| 1468 | ); |
| 1469 | |
| 1470 | return CalcShapValuesWithPreparedTrees( |
| 1471 | model, |
| 1472 | dataset, |
| 1473 | fixedFeatureParams, |
| 1474 | logPeriod, |
| 1475 | preparedTrees, |
| 1476 | localExecutor, |
| 1477 | calcType |
| 1478 | ); |
| 1479 | } |
| 1480 | |
| 1481 | TVector<TVector<double>> CalcShapValues( |
| 1482 | const TFullModel& model, |
no test coverage detected