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

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

Usage no npm install needed!

<script type="module">
  import pipcookPluginsImageClassificationTfjsModelTrain from 'https://cdn.skypack.dev/@pipcook/plugins-image-classification-tfjs-model-train';
</script>

README

@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
  }
}