Krotos Modules 3
Loading...
Searching...
No Matches
krotos::SentenceTransformer Class Reference

#include <SentenceTransformer.h>

Public Member Functions

 SentenceTransformer ()
 
 ~SentenceTransformer ()
 
std::vector< float > encode (std::string sentence) const
 

Private Member Functions

File getModelPath () const
 
void redirectPrint (File filePath, int &stdoutFd)
 
void restorePrint (int stdoutFd)
 

Private Attributes

bool m_modelFileAvailable {false}
 
bert_ctx * bctx
 
int n_max_tokens = 0
 

Constructor & Destructor Documentation

◆ SentenceTransformer()

krotos::SentenceTransformer::SentenceTransformer ( )

◆ ~SentenceTransformer()

krotos::SentenceTransformer::~SentenceTransformer ( )

Member Function Documentation

◆ encode()

std::vector< float > krotos::SentenceTransformer::encode ( std::string sentence) const

◆ getModelPath()

File krotos::SentenceTransformer::getModelPath ( ) const
private

◆ redirectPrint()

void krotos::SentenceTransformer::redirectPrint ( File filePath,
int & stdoutFd )
private

◆ restorePrint()

void krotos::SentenceTransformer::restorePrint ( int stdoutFd)
private

Member Data Documentation

◆ bctx

bert_ctx* krotos::SentenceTransformer::bctx
private

◆ m_modelFileAvailable

bool krotos::SentenceTransformer::m_modelFileAvailable {false}
private

◆ n_max_tokens

int krotos::SentenceTransformer::n_max_tokens = 0
private

The documentation for this class was generated from the following files: