Nodes/ComfyUI Neural Network Toolkit NNT /NNT Dataset To Target Tensor
ComfyUI Node

NNT Dataset To Target Tensor

A ComfyUI node in NNT Neural Network Toolkit/Data Processing with 15 inputs and 4 outputs.

By inventorado·Created 2 years ago·Updated 2 years ago· 69
NNT Dataset To Target Tensor
  • dataset
  • target_tensor
  • info
  • label_info
  • collated_outputs
target_columnlabel
target_typeclassification
num_classes10
use_data_collatorTrue
paddingmax_length
pad_to_multiple_of8
return_tensorspt
encodingsparse
label_smoothing0.10
create_label_mapsTrue
custom_label_map{}
detach_tensorTrue
requires_gradFalse
make_cloneTrue
CategoryNNT Neural Network Toolkit/Data Processing

Inputs (15)

NameTypeDefaultDescription
datasetDATASET
target_columnSTRINGlabel
target_typeCOMBOclassification3 options: classification, regression, multi_label
num_classesINT102–1000
use_data_collatorCOMBOTrue2 options: True, False
paddingCOMBOmax_length3 options: max_length, longest, do_not_pad
pad_to_multiple_ofINT81–128
return_tensorsCOMBOpt2 options: pt, tf
encodingCOMBOsparse3 options: sparse, one_hot, label_smooth
label_smoothingFLOAT0.100–0.5
create_label_mapsCOMBOTrue2 options: True, False
custom_label_mapSTRING{}
detach_tensorCOMBOTrue2 options: True, False
requires_gradCOMBOFalse2 options: True, False
make_cloneCOMBOTrue2 options: True, False

Outputs (4)

NameTypeDescription
target_tensorTENSOR
infoSTRING
label_infoDICT
collated_outputsDICT