Exemple #1
0
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
Exemple #2
0
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