HowToSD-ComfyUI-Pt-Wrapper/docs/reference/ptn_hf_fine_tuned_classification_model.md

Ptn Hf Fine Tuned Classification Model

A binary classification model containing a Hugging Face pretrained Transformer model.

    Args:
    model_name (str): Name or path of the pretrained Hugging Face model.
    use_mean_pooling (bool): If True, average token embeddings using the attention mask.
    Otherwise, use pooler output or CLS token embedding.
    dropout (float): Dropout rate applied before the final linear layer.
                     Set to 0.0 to disable dropout. Must be ≥ 0.

Input

Name	Data type
model_name	String
use_mean_pooling	Boolean
dropout	Float

Output

Data type
Ptmodel

---

Category: PyTorch wrapper - Model

ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.