Skip to content

targettransformercls

targettransformercls

Module for TargetTransformerClassification

Classes

TargetTransformerClassification

TargetTransformerClassification()

Bases: NetTargetTransformer

NetTargetTransformer for Classification

Source code in niceml/mlcomponents/targettransformer/targettransformercls.py
def __init__(self):
    super().__init__()
    self.use_binary: bool = False
    self.multi_label_binarizer = None

Functions