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

$3

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

$3

| Parameter | Type | Description |
|:----------|:-----|:------------|

$3

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

$3

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