TextEmbedder

TextEmbedder i klasës përfundimtare publike

Kryen nxjerrjen e ngulitur në tekst.

Ky API pret një model TFLite me Metadata të Modelit TFLite (opsionale).

Metadatat kërkohen për modelet me tensorë të hyrjes int32 sepse përmbajnë njësinë e procesit të hyrjes për Tokenizuesin e modelit. Nuk kërkohen meta të dhëna për modelet me tensorë të hyrjes së vargut.

  • Tenzoret hyrëse
    • Tre tensorë hyrës ( kTfLiteInt32 ) të formës [batch_size x bert_max_seq_len] që përfaqësojnë ID-të e hyrjes, ID-të e maskave dhe ID-të e segmentit. Ky nënshkrim i hyrjes kërkon një njësi procesi Bert Tokenizer në meta të dhënat e modelit.
    • Ose një tensor hyrës ( kTfLiteInt32 ) me formë [batch_size x max_seq_len] që përfaqëson ID-të e hyrjes. Ky nënshkrim hyrës kërkon një njësi procesi Regex Tokenizer në meta të dhënat e modelit.
    • Ose një tensor hyrës ( kTfLiteString ) që është pa formë ose ka formën [1] që përmban vargun hyrës.
  • Të paktën një tensor dalës ( kTfLiteFloat32 / kTfLiteUint8 ) me formë [1 x N] ku N është numri i dimensioneve në ngulitje të prodhuara.

Klasat e mbivendosura

klasës TextEmbedder.TextEmbedderOptions Opsionet për konfigurimin e një TextEmbedder .

Metodat publike

i pavlefshëm
mbyll ()
Mbyll dhe pastron TextEmbedder -in.
dyfishtë statike
kosinusNgjashmëria ( Embedding u, Embedding v)
Funksioni i dobisë për të llogaritur ngjashmërinë e kosinusit midis dy objekteve Embedding .
TextEmbedder statik
createFromFile (konteksti i kontekstit, modeli i vargut shteg)
Krijon një shembull TextEmbedder nga një skedar modeli dhe TextEmbedder.TextEmbedderOptions të paracaktuar.
TextEmbedder statik
createFromFile (konteksti i kontekstit, modeli i skedaritFile )
Krijon një shembull TextEmbedder nga një skedar modeli dhe TextEmbedder.TextEmbedderOptions të paracaktuar.
TextEmbedder statik
createFromOptions (konteksti i kontekstit, opsionet e TextEmbedder.TextEmbedderOptions )
Krijon një shembull TextEmbedder nga TextEmbedder.TextEmbedderOptions .
TextEmbedderResult
embed (Teksti i hyrjes së vargut )
Kryen ekstraktimin e integruar në tekstin hyrës.

Metodat e trashëguara

Metodat publike

mbyllje boshllëku publik ()

Mbyll dhe pastron TextEmbedder -in.

kosinusi i dyfishtë statik publik Ngjashmëria ( Embedding u, Embedding v)

Funksioni i dobisë për të llogaritur ngjashmërinë e kosinusit midis dy objekteve Embedding .

Parametrat
u
v
Hedhjet
IllegalArgumentPërjashtim nëse futjet janë të llojeve të ndryshme (float kundrejt kuantizuar), kanë madhësi të ndryshme ose kanë një normë L2 prej 0.

publik statik TextEmbedder createFromFile (konteksti i kontekstit, modeli i vargut Path)

Krijon një shembull TextEmbedder nga një skedar modeli dhe TextEmbedder.TextEmbedderOptions të paracaktuar.

Parametrat
kontekst një ERROR(/Context) në Android.
modelRruga rrugën drejt modelit të tekstit me meta të dhëna në asete.
Hedhjet
nëse ka ndonjë gabim gjatë krijimit të TextEmbedder .

publik statik TextEmbedder createFromFile (konteksti i kontekstit, modeli i skedaritFile )

Krijon një shembull TextEmbedder nga një skedar modeli dhe TextEmbedder.TextEmbedderOptions të paracaktuar.

Parametrat
kontekst një ERROR(/Context) në Android.
modelFile modeli i tekstit File skedarit.
Hedhjet
Përjashtim IO nëse ndodh një gabim I/O kur hapet skedari i modelit tflite.
nëse ka një gabim gjatë krijimit të TextEmbedder .

publike statike TextEmbedder createFromOptions (konteksti i kontekstit, opsionet e TextEmbedder.TextEmbedderOptions )

Krijon një shembull TextEmbedder nga TextEmbedder.TextEmbedderOptions .

Parametrat
kontekst një ERROR(/Context) në Android.
opsionet një shembull TextEmbedder.TextEmbedderOptions .
Hedhjet
nëse ka një gabim gjatë krijimit të TextEmbedder .

TextEmbedder Rezultat publik ngulitje ( String inputText)

Kryen ekstraktimin e integruar në tekstin hyrës.

Parametrat
inputText një String për përpunim.