/pipcook-plugin-tfjs-image-classification-model-train

The Pipcook model train plugin for tfjs-based image classification tasks.

Primary LanguageTypeScript

@pipcook/plugins-image-classification-tfjs-model-train

Description

This plugin will be used to train image classification models built on top of Tensorflow.js.

Required Parameters

Parameter Type Description

Optional Parameters

Parameter Type Default Value Description
epochs number 10 how many epochs to train
batchSize number 16 data's batch size to be trained for every iteration

Example

"modelTrain": {
  "package": "@pipcook/plugins-image-classification-tfjs-model-train",
  "params": {
    "epochs": 15
  }
}