o pi@s<ddlZddlZddlmZddlmZGdddeZdS)N)loss_and_miner_utils) BaseMinercsNeZdZdfdd ZddZddZdd Zd d Zd d ZddZ Z S)HDCMiner?c s4tjdi|||_|jdgdd|dS)Nfilter_percentageF) list_of_namesis_stat)super__init__radd_to_recordable_attributes reset_idx)selfrkwargs __class__r \/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/miners/hdc_miner.pyr s  zHDCMiner.__init__cCs|||}||||j|jf|j|jfdD]5\}\}}t|dkrP|||f} t| } tt |j | } | |} t j| | | d\} }|||q|j|j|j|jfS)N)posnegr)klargest)distanceset_idxa1pa2nitemslenintmathceilrshould_select_largesttorchtopkfilter_original_indices)r embeddingslabelsref_emb ref_labelsmatnameanchorotherpairs num_pairsrr_idxr r rmines        z HDCMiner.minecCs(|jjr |dkr dSdS|dkrdSdS)NrFT)r is_inverted)rr-r r rr$&szHDCMiner.should_select_largestcCs*|jst||\|_|_|_|_dSdS)N)was_set_externallylmuget_all_pairs_indicesrrrr)rr)r+r r rr+s zHDCMiner.set_idxcCs&t||\|_|_|_|_d|_dS)NT)r7convert_to_pairsrrrrr6)rexternal_indices_tupler)r r rset_idx_externally1s zHDCMiner.set_idx_externallycCsd\|_|_|_|_d|_dS)N)NNNNF)rrrrr6)rr r rr7s zHDCMiner.reset_idxcCs@|dkr|j||_|j||_dS|j||_|j||_dS)Nr)rrrr)rr-r3r r rr';s   z HDCMiner.filter_original_indices)r) __name__ __module__ __qualname__r r4r$rr;rr' __classcell__r r rrr sr)r"r%utilsrr7 base_minerrrr r r rs