predict method

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

Implementation

List<List<double>> predict(List<List<int>> X) {
  final xs = X.map((toks) => _embedAndPool(toks)).toList();
  return head.predict(xs);
}