https://github.com/imgcook/pipcook-plugin-tfjs-image-classification-model-train

The Pipcook model train plugin for tfjs-based image classification tasks.
https://github.com/imgcook/pipcook-plugin-tfjs-image-classification-model-train

Last synced: about 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/imgcook/pipcook-plugin-tfjs-image-classification-model-train
• Owner: imgcook
• Created: 2020-08-13T05:46:32.000Z (about 4 years ago)
• Default Branch: master
• Last Pushed: 2021-02-23T05:57:40.000Z (over 3 years ago)
• Last Synced: 2024-04-16T20:06:51.304Z (5 months ago)
• Language: TypeScript
• Size: 8.79 KB
• Stars: 0
• Watchers: 6
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md

Awesome Lists containing this project

• awesome-imgcook - imgcook/pipcook-plugin-tfjs-image-classification-model-train - based image classification model. (Pipcook Plugins / Model Train)

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

```json

"modelTrain": {

  "package": "@pipcook/plugins-image-classification-tfjs-model-train",

  "params": {

    "epochs": 15

  }

}

```