ComfyUI Node
Pt Predict Classification Model
A ComfyUI node in Training with 4 inputs and 3 outputs.
Pt Predict Classification Model
- model
- inputs
- class_id_to_name_map
- class_name
- class_id
- probability
◄use_gpufalse►
CategoryTraining
Inputs (4)
| Name | Type | Default | Description |
|---|---|---|---|
| model | PTMODEL | — | |
| inputs | TENSOR | — | |
| class_id_to_name_map | PYDICT | — | |
| use_gpu | BOOLEAN | false | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| class_name | STRING | — |
| class_id | INT | — |
| probability | FLOAT | — |