logisticRegressionPredictClass function

int logisticRegressionPredictClass(
  1. List<double> weights,
  2. List<double> features, {
  3. double threshold = 0.5,
})

Implementation

int logisticRegressionPredictClass(
  List<double> weights,
  List<double> features, {
  double threshold = 0.5,
}) {
  var z = weights[0];
  for (var i = 0; i < features.length && i + 1 < weights.length; i++) {
    z += weights[i + 1] * features[i];
  }
  return _sigmoid(z) >= threshold ? 1 : 0;
}