This plugin will be used to train image classification models built on top of Tensorflow.js.
| Parameter | Type | Description |
|---|
| 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 |
"modelTrain": {
"package": "@pipcook/plugins-image-classification-tfjs-model-train",
"params": {
"epochs": 15
}
}