predict method

List<int> predict(List<List<double>> X)

Implementation

List<int> predict(List<List<double>> X) {
  if (centroids == null) throw StateError('Model not fitted');
  return X.map((x) {
    var best = 0;
    var bestD = double.infinity;
    for (var j = 0; j < centroids!.length; j++) {
      final d = _distSq(x, centroids![j]);
      if (d < bestD) {
        bestD = d;
        best = j;
      }
    }
    return best;
  }).toList();
}