findCoeff function

dynamic findCoeff(
  1. dynamic train,
  2. dynamic l_rate,
  3. dynamic n_epoch
)

Implementation

findCoeff(var train, l_rate, n_epoch) {
  List coeff = [];
  for (int i = 0; i < train[0].length; i++) {
    //generator function
    coeff.add(0.0);
  }

  for (int i = 0; i < n_epoch; i++) {
    for (int y = 0; y < train.length - 1; y++) {
      var yhat = log_reg(train[y], coeff);
      List temp = train[y];
      var error = train[y][temp.length - 1] - yhat;
      coeff[0] = coeff[0] + l_rate * error * yhat * (1.0 - yhat);
      for (int z = 0; z < train[y].length - 1; z++) {
        coeff[z + 1] =
            coeff[z + 1] + l_rate * error * yhat * (1.0 - yhat) * train[y][z];
      }
    }
  }
  return coeff;
}