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