@iSrSSKrSSKJrJrJrJr SSKrSSK J r SSK J r \R"\5r"SS5rg)z8Dataset for contrastive learning with triplet structure.N)ListTupleDictAny) MemoryStore)PatternEmbedderc\rSrSrSrSS\S\S\4SjjrS\ R4Sjr S\ \ \ R\ R\ R44S jrS\4S jrS \S\ \ R\ R\ R44S jrS rg)PreferenceDataset zDataset for triplet-based contrastive learning. Each sample is (anchor, positive, negative): - anchor: User preference vector - positive: Liked pattern vector - negative: Disliked pattern vector memory_storeembedderuser_idcXlX lX0lUR5UlURR U5UlUR c([RSUS35 /Ul /Ul OUR R S/5Vs/sHn[R"U5PM snUl UR R S/5Vs/sHn[R"U5PM snUl [RS[UR5S[UR5S35 gs snfs snf) zInitialize preference dataset. Args: memory_store: Memory store instance embedder: Pattern embedder for encoding patterns user_id: User identifier NzUser z not found in memorypositivenegativezDataset initialized: z positive, z negative vectors)r r rload_user_memory user_memoryget user_profileloggerwarningpositive_vectorsnegative_vectorsnparrayinfolen)selfr r rvs C/home/ubuntu/codebase/yexijia/保研/iCoLoc/src/learning/dataset.py__init__PreferenceDataset.__init__s%)  (88: ,,009    $ NNU7)+?@ A$&D !$&D !;?:K:K:O:OPZ\^:_$`:_QRXXa[:_$`D !:>:K:K:O:OPZ\^:_$`:_QRXXa[:_$`D ! +C0E0E,F+G{$//011BD E%a$`s  E ! Ereturnc[UR5S:XaS[UR5S:Xa:URRR 5n[ R"U5$[UR5S:a [ R"URSS9nOEUR(aURSRSOSn[ R"U5n[UR5S:a [ R"URSS9nOEUR(aURSRSOSn[ R"U5nX#- nU$)zUBuild user anchor vector. Returns: User preference vector (mean_pos - mean_neg) r)axisi) rrrr model get_sentence_embedding_dimensionrzerosmeanshape)rdimmean_posmean_neg user_vectors r build_user_vector#PreferenceDataset.build_user_vector2s t$$ % *s43H3H/IQ/N--%%FFHC88C=  t$$ % )wwt441=H7;7L7L$''*003RUCxx}H t$$ % )wwt441=H7;7L7L$''*003RUCxx}H) cr[UR5S:Xd[UR5S:Xa[R S5 /$UR 5n/nURH)nURHnUR XU45 M M+ [RS[U5S35 U$)zVGenerate triplet samples. Returns: List of (anchor, positive, negative) triplets rz(Insufficient data for triplet generationz Generated z triplets)rrrrrr/appendr)ranchortripletspos_vecneg_vecs r get_tripletsPreferenceDataset.get_tripletsPs t$$ % *c$2G2G.HA.M NNE FI''),,G00' :;1-  jXy9:r1c8UR5n[U5$)zGet dataset size.)r8r)rr5s r __len__PreferenceDataset.__len__es$$&8}r1idxc|UR5nU[U5:a[SUS[U535eX!$)zjGet a triplet sample. Args: idx: Sample index Returns: (anchor, positive, negative) triplet zIndex z" out of range for dataset of size )r8r IndexError)rr=r5s r __getitem__PreferenceDataset.__getitem__jsD$$& #h- vcU*LSQY]O\] ]}r1)r r rrrrrN)user_001)__name__ __module__ __qualname____firstlineno____doc__rrstrr!rndarrayr/rrr8intr;r@__static_attributes__r1r r r sE[EOEVYE82::<d5RZZ)K#LM* s uRZZRZZ-O'P r1r )rGnumpyrtypingrrrrlogging memory.storerlearning.embedderr getLoggerrCrr rLr1r rSs6>))$-   8 $jjr1