findCoeff function
dynamic
findCoeff(
- dynamic train,
- dynamic l_rate,
- 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;
}