def testdata_match_kernel_L0():
from ibeis.algo.hots.smk import smk_debug
from ibeis.algo.hots import hstypes
np.random.seed(0)
smk_alpha = 3.0
smk_thresh = 0.0
num_qrvecs_per_word = [0, 1, 3, 4, 5]
num_drvecs_per_word = [0, 1, 2, 4, 6]
qrvecs_list = [
smk_debug.get_test_rvecs(n, dim=2) for n in num_qrvecs_per_word
]
drvecs_list = [
smk_debug.get_test_rvecs(n, dim=2) for n in num_drvecs_per_word
]
daids_list = [list(range(len(rvecs))) for rvecs in drvecs_list]
qaids_list = [[42] * len(rvecs) for rvecs in qrvecs_list]
qmaws_list = [smk_debug.get_test_maws(rvecs) for rvecs in qrvecs_list]
dmaws_list = [
np.ones(rvecs.shape[0], dtype=hstypes.FLOAT_TYPE)
for rvecs in drvecs_list
]
idf_list = [1.0 for _ in qrvecs_list]
daid2_sccw = {daid: 1.0 for daid in range(10)}
query_sccw = smk_scoring.sccw_summation(qrvecs_list, idf_list, qmaws_list,
smk_alpha, smk_thresh)
qaid2_sccw = {42: query_sccw}
core1 = smk_alpha, smk_thresh, query_sccw, daids_list, daid2_sccw
core2 = qrvecs_list, drvecs_list, qmaws_list, dmaws_list, idf_list
extra = qaid2_sccw, qaids_list
return core1, core2, extra
def testdata_match_kernel_L0():
from ibeis.algo.hots.smk import smk_debug
from ibeis.algo.hots import hstypes
np.random.seed(0)
smk_alpha = 3.0
smk_thresh = 0.0
num_qrvecs_per_word = [0, 1, 3, 4, 5]
num_drvecs_per_word = [0, 1, 2, 4, 6]
qrvecs_list = [smk_debug.get_test_rvecs(n, dim=2) for n in num_qrvecs_per_word]
drvecs_list = [smk_debug.get_test_rvecs(n, dim=2) for n in num_drvecs_per_word]
daids_list = [list(range(len(rvecs))) for rvecs in drvecs_list]
qaids_list = [[42] * len(rvecs) for rvecs in qrvecs_list]
qmaws_list = [smk_debug.get_test_maws(rvecs) for rvecs in qrvecs_list]
dmaws_list = [np.ones(rvecs.shape[0], dtype=hstypes.FLOAT_TYPE) for rvecs in drvecs_list]
idf_list = [1.0 for _ in qrvecs_list]
daid2_sccw = {daid: 1.0 for daid in range(10)}
query_sccw = smk_scoring.sccw_summation(qrvecs_list, idf_list, qmaws_list, smk_alpha, smk_thresh)
qaid2_sccw = {42: query_sccw}
core1 = smk_alpha, smk_thresh, query_sccw, daids_list, daid2_sccw
core2 = qrvecs_list, drvecs_list, qmaws_list, dmaws_list, idf_list
extra = qaid2_sccw, qaids_list
return core1, core2, extra
def testdata_similarity_function():
from ibeis.algo.hots.smk import smk_debug
qrvecs_list = [smk_debug.get_test_rvecs(_) for _ in range(10)]
drvecs_list = [smk_debug.get_test_rvecs(_) for _ in range(10)]
return qrvecs_list, drvecs_list
def testdata_similarity_function():
from ibeis.algo.hots.smk import smk_debug
qrvecs_list = [smk_debug.get_test_rvecs(_) for _ in range(10)]
drvecs_list = [smk_debug.get_test_rvecs(_) for _ in range(10)]
return qrvecs_list, drvecs_list
