\iLSrSSKJrJr SSKrSSKJr SSKJ r "SS5r g)z2Pattern embedding module for vectorizing patterns.)ListUnionN)SentenceTransformer)Pathc\rSrSrSrS S\4SjjrS\\S\R4Sjr S\\\S\R4S jr S\ 4S jr S rg )PatternEmbedder z4Pattern embedder for converting patterns to vectors. model_namec[[5R5RRRS- S- n[ [ U55Ulg)z[Initialize pattern embedder. Args: model_name: Name of the sentence transformer model modelszall-MiniLM-L6-v2N)r__file__resolveparentrstrmodel)selfr _MODEL_DIRs D/home/ubuntu/codebase/yexijia/保研/iCoLoc/src/learning/embedder.py__init__PatternEmbedder.__init__ sC(^++-44;;BBXMPbb (Z9 patternreturncZSRU5nURRU5$)zEncode a single pattern to vector. Args: pattern: List of feature types in the pattern Returns: Vector representation of the pattern  joinrencode)rrtexts rencode_patternPatternEmbedder.encode_patterns'xx zz  &&rpatternscUVs/sHnSRU5PM nnURRU5$s snf)zEncode multiple patterns to vectors. Args: patterns: List of patterns, each is a list of feature types Returns: Array of vectors, shape (n_patterns, embedding_dim) rr)rr"ptextss rencode_patternsPatternEmbedder.encode_patterns&s:'//h!h/zz  ''0s=c6URR5$)zLGet the dimension of sentence embeddings. Returns: Embedding dimension )r get_sentence_embedding_dimension)rs rr)0PatternEmbedder.get_sentence_embedding_dimension2s zz::< :3 : 'd3i 'BJJ ' (T#Y (BJJ (=#=rr) r/typingrrnumpyr0sentence_transformersrpathlibrrr4rrr9s85.=.=r