Function svm_train_one

ml/src/svm.cpp:1910–1967

};

static decision_function
svm_train_one(const svm_problem* prob, const svm_parameter* param, double Cp, double Cn)
{
  double* alpha = Malloc(double, prob->l);
  Solver::SolutionInfo si;

  switch (param->svm_type) {

  case C_SVC:
    solve_c_svc(prob, param, alpha, &si, Cp, Cn);
    break;

  case NU_SVC:
    solve_nu_svc(prob, param, alpha, &si);
    break;

  case ONE_CLASS:
    solve_one_class(prob, param, alpha, &si);
    break;

  case EPSILON_SVR:
    solve_epsilon_svr(prob, param, alpha, &si);
    break;

  case NU_SVR:
    solve_nu_svr(prob, param, alpha, &si);
    break;
  }

  info("obj = %f, rho = %f\n", si.obj, si.rho);

  // output SVs

  int nSV = 0;
  int nBSV = 0;

  for (int i = 0; i < prob->l; i++) {
    if (std::abs(alpha[i]) > 0) {
      ++nSV;

      if (prob->y[i] > 0) {
        if (std::abs(alpha[i]) >= si.upper_bound_p)
          ++nBSV;
      }
      else {
        if (std::abs(alpha[i]) >= si.upper_bound_n)
          ++nBSV;
      }
    }
  }

  info("nSV = %d, nBSV = %d\n", nSV, nBSV);

  decision_function f;
  f.alpha = alpha;
  f.rho = si.rho;
  return f;
}