ComfyUI Node: NNT Dataset To Target Tensor

Authored by inventorado

Created

Updated

67 stars

Category

NNT Neural Network Toolkit/Data Processing

Inputs

dataset DATASET
target_column STRING
target_type
  • classification
  • regression
  • multi_label
num_classes INT
use_data_collator
  • True
  • False
padding
  • max_length
  • longest
  • do_not_pad
pad_to_multiple_of INT
return_tensors
  • pt
  • tf
encoding
  • sparse
  • one_hot
  • label_smooth
label_smoothing FLOAT
create_label_maps
  • True
  • False
custom_label_map STRING
detach_tensor
  • True
  • False
requires_grad
  • True
  • False
make_clone
  • True
  • False

Outputs

TENSOR

STRING

DICT

DICT

Extension: ComfyUI Neural Network Toolkit NNT

Neural Network Toolkit (NNT) for ComfyUI is an extensive set of custom ComfyUI nodes for designing, training, and fine-tuning neural networks. This toolkit allows defining models, layers, training workflows, transformers, and tensor operations in a visual manner using nodes.

Authored by inventorado

View Nodes

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more