def _load_embs(zq_emb_lst=None, az_emb_lst=None): ALL_EMB_MATRIX_READY_DICT = {} # ################################################################################# # LOAD ZQ EMB # ################################################################################# EBL = EmbBatchLoader(all_emb_cols=EMB_keys2do, key2index=KEY2INDEX_DICT) for ii, i_emb in enumerate(zq_emb_lst, 1): ALL_EMB_MATRIX_READY_DICT[ f'emb_matrix_ready_dict_{ii}'] = EBL.get_batch_emb_matrix( marker=i_emb) return ALL_EMB_MATRIX_READY_DICT
def _load_embs(zq_emb_lst=None, az_emb_lst=None): ''' :param zq_emb_lst: embedding files group name :param az_emb_lst: all embedding in same path so useless :return: ''' ALL_EMB_MATRIX_READY_DICT = {} # ################################################################################# # LOAD ZQ EMB # ################################################################################# EBL = EmbBatchLoader(all_emb_cols=EMB_keys2do, key2index=KEY2INDEX_DICT) for ii, i_emb in enumerate(zq_emb_lst, 1): ALL_EMB_MATRIX_READY_DICT[ f'emb_matrix_ready_dict_{ii}'] = EBL.get_batch_emb_matrix( marker=i_emb) return ALL_EMB_MATRIX_READY_DICT