await model.fit(
  xData, yData, {
    batchSize: batchSize,
    epochs: epochs
});