fit method

void fit(
  1. List<List<int>> X,
  2. List<List<double>> Y, {
  3. int? batchSize,
  4. bool verbose = false,
})

Implementation

void fit(
  List<List<int>> X,
  List<List<double>> Y, {
  int? batchSize,
  bool verbose = false,
}) {
  final xs = X.map((toks) => _embedAndPool(toks)).toList();
  head.fit(xs, Y, batchSize: batchSize, verbose: verbose);
}