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

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

Last synced: 3 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/imgcook/pipcook-plugin-tensorflow-image-classification-model-train
• Owner: imgcook
• License: apache-2.0
• Created: 2020-08-13T05:42:18.000Z (over 4 years ago)
• Default Branch: master
• Last Pushed: 2020-10-30T07:10:04.000Z (about 4 years ago)
• Last Synced: 2024-08-02T19:01:36.636Z (4 months ago)
• Language: TypeScript
• Size: 18.6 KB
• Stars: 0
• Watchers: 6
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

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

README

        
# @pipcook/plugins-image-classification-tensorflow-model-train

### Description

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

### 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-tensorflow-model-train",

  "params": {

    "epochs": 15

  }

}

```