Class: Transformers::ClassificationFunction

Inherits:
ExplicitEnum show all
Defined in:
lib/transformers/pipelines/image_classification.rb

Constant Summary collapse

SIGMOID =
"sigmoid"
SOFTMAX =
"softmax"
NONE =
"none"

Method Summary

Methods inherited from ExplicitEnum

#initialize, #to_s

Constructor Details

This class inherits a constructor from Transformers::ExplicitEnum