Function BinaryClassificationTarget

src/model/encoder/target_encoder.cc:18–35

namespace embedx {

std::vector<GraphNode*> BinaryClassificationTarget(const std::string& prefix,
                                                   GraphNode* X, GraphNode* Y,
                                                   int has_w) {
  DXCHECK_THROW(X->shape().is_rank(2));
  DXCHECK_THROW(X->shape()[1] == 1);
  DXCHECK_THROW(Y->shape()[1] == 1);
  auto* L = deepx_core::SigmoidBCELoss(prefix + "L", X, Y);
  auto* P = deepx_core::Sigmoid(prefix + "P", X);
  if (has_w) {
    auto* W = deepx_core::GetW(1);
    auto* WL = deepx_core::Mul(prefix + "WL", L, W);
    auto* WM = deepx_core::ReduceMean(prefix + "WM", WL);
    return {WM, P};
  } else {
    auto* M = deepx_core::ReduceMean(prefix + "M", L);
    return {M, P};
  }
}

std::vector<GraphNode*> BinaryClassificationTarget(GraphNode* X, GraphNode* Y,
                                                   int has_w) {