Train the CNN Model in TFJS

tfjs-blog-train-model.js

class VisionModelWorker {

   ... // create and getData functions
   
    async train(epochs, batchSize) {
        this.model.compile({ optimizer: tf.train.adam(), loss: 'categoricalCrossentropy', metrics: ['accuracy'] });
        const historyObj = await this.model.fit(this.dataBunch.trainX, this.dataBunch.trainY, {
            batchSize: batchSize,
            validationData: [this.dataBunch.testX, this.dataBunch.testY],
            epochs: epochs,
            shuffle: true
        });
        this.trainingHistories.push(historyObj);
        console.debug(`Training history: ${JSON.stringify(historyObj.history)}`);
        return historyObj;
    }
}