c

com.johnsnowlabs.ml.onnx

OnnxSetFitClassifier

class OnnxSetFitClassifier extends AnyRef

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OnnxSetFitClassifier
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new OnnxSetFitClassifier(embeddingsModel: OnnxWrapper, classifierModel: OnnxWrapper, tokenizer: SetFitTokenizer, labels: Map[Int, String], hasDifferentiableHead: Boolean = false)

Value Members

  1. val classifierModel: OnnxWrapper
  2. val embeddingsModel: OnnxWrapper
  3. val hasDifferentiableHead: Boolean
  4. val labels: Map[Int, String]
  5. def predict(documents: Seq[Annotation], batchSize: Int, maxSeqLength: Int, caseSensitive: Boolean): Seq[(String, Map[String, Float])]
  6. val tokenizer: SetFitTokenizer