iTSSKrSSKrSrSrSrSrSrS SjrSr S r S r S r g) Ncn[R"U5nURU5n[R"USS9nS=pEUR U5nUS:a5[R "[R"U5XCR5 URU5nUR[RR:wa![R"X R4SS9nO8s5    -C ;;rttRWWbgg ..r4c[R"UR5n[R"UR5R UR UR 5nX!4$)zfextract matrix + bias from the PCA object works for any linear transform (OPQ, random rotation, etc.) )r r6bAr8d_outd_in)pcar@rAs r$get_LinearTransform_matrixrE>sH cee$A cee$,,SYYAA 4Kr4cRURup#UbURU4:Xde[R"X2USL5n[R"UR 5UR 5 Ub [R"XR 5 SUlUR5 U$)z@make a linear transform from a matrix and a bias term (optional)NT) shaper LinearTransformcopy_array_to_vectorravelrAr@ is_trainedset_is_orthonormal)rAr@rBrClts r$make_LinearTransform_matrixrNGs''KE}ww5)###   tATM :B qwwy"$$/} ""1dd+BM Ir4c [R"UR5RSUR5n[R"UR 5n[ UR5Vs/sHnXUX#S-PM sn$s snf)z-return to codebooks of an additive quantizer r)r r6 codebooksr8dcodebook_offsetsr(r9)aqrQcor*s r$ get_additive_quantizer_codebooksrVUs{%%bll3;;BEI   r22 3Brtt A Q%q5 "  s2B c[R"UR5RS5nUR UR UR 5$)z0copy and return the data matrix in an IndexFlat float32)r r6rviewr8ntotalrR)indexxbs r$ get_flat_datar]_s:   u{{ + 0 0 ;B ::ellEGG ,,r4c[R"UR5RURUR 5$)z1get the codes from an indexFlatCodes as an array )r r6rr8rZr) index_flats r$get_flat_codesr`es7  !1!1 2 : ::// 11r4cUR5n[R"URUR4SS9n[ R "[ R"U5URUR5 U$)zaget the neighbor list for the vectors stored in the NSG structure, as a N-by-K matrix of indices int32r) get_final_graphr r NKr rrdatar)nsggraph neighborss r$get_NSG_neighborsrjks_    !E%''577+7;I LL y!  r4)N) numpyr r r%r+r3r>rErNrVr]r`rjr4r$rms?   >0/  - 1 r4