case class Gliner2ProcessedEmbeddings(tokenEmbeddings: Array[Array[Float]], schemaEmbeddings: Array[Array[Array[Float]]], textTokens: Array[String], schemaTokensList: Array[Array[String]], taskTypes: Array[String], startMapping: Array[Int], endMapping: Array[Int], originalText: String, originalSchema: Gliner2Schema, sampleIndex: Int) extends Product with Serializable
Per-sample embeddings extracted from encoder output. Splits aggregated token and schema embeddings and returns them separately.
Python Reference: _extract_embeddings_from_batch().
- tokenEmbeddings
Word-level text embeddings (text_len, 768)
- schemaEmbeddings
Schema embeddings per task (num_schemas, num_tokens, 768)
- textTokens
Original text tokens
- schemaTokensList
Schema tokens per task
- taskTypes
Task types (e.g., "entities", "classifications")
- startMapping
Character start positions for tokens
- endMapping
Character end positions for tokens
- originalText
Original text string
- originalSchema
Original schema dictionary
- sampleIndex
Index of this sample in the batch
- Alphabetic
- By Inheritance
- Gliner2ProcessedEmbeddings
- Serializable
- Serializable
- Product
- Equals
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Instance Constructors
-
new
Gliner2ProcessedEmbeddings(tokenEmbeddings: Array[Array[Float]], schemaEmbeddings: Array[Array[Array[Float]]], textTokens: Array[String], schemaTokensList: Array[Array[String]], taskTypes: Array[String], startMapping: Array[Int], endMapping: Array[Int], originalText: String, originalSchema: Gliner2Schema, sampleIndex: Int)
- tokenEmbeddings
Word-level text embeddings (text_len, 768)
- schemaEmbeddings
Schema embeddings per task (num_schemas, num_tokens, 768)
- textTokens
Original text tokens
- schemaTokensList
Schema tokens per task
- taskTypes
Task types (e.g., "entities", "classifications")
- startMapping
Character start positions for tokens
- endMapping
Character end positions for tokens
- originalText
Original text string
- originalSchema
Original schema dictionary
- sampleIndex
Index of this sample in the batch
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
- val endMapping: Array[Int]
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
def
numSchemas: Int
Number of schemas in this sample.
- val originalSchema: Gliner2Schema
- val originalText: String
- val sampleIndex: Int
- val schemaEmbeddings: Array[Array[Array[Float]]]
- val schemaTokensList: Array[Array[String]]
- val startMapping: Array[Int]
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
- val taskTypes: Array[String]
-
def
textLength: Int
Number of text tokens.
- val textTokens: Array[String]
- val tokenEmbeddings: Array[Array[Float]]
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()