do_test.py 文件源码

def setup(dataset = 'proton-beam', n = 1000, ngold = 0, rand_shuffle = None):
    start.main(dataset)

    if rand_shuffle != None:
        random.shuffle(start.turk_data_id, lambda : rand_shuffle)
        random.shuffle(start.rel, lambda : rand_shuffle)

    lc_gold = crowd_model.labels_collection(start.turk_data_id, start.rel)
    gold_dic = lc_gold.get_true_ss()

    lc1 = crowd_model.labels_collection(start.turk_data_id[:n], start.rel[:ngold] + (n-ngold)*[None])
    tc = crowd_model.tc_model(lc1)

    lc2 = crowd_model.labels_collection(start.turk_data_id[:n], start.rel[:ngold] + (n-ngold)*[None])
    mv = crowd_model.mv_model(lc2)

    lc3 = crowd_model.labels_collection(start.turk_data_id[:n], start.rel[:ngold] + (n-ngold)*[None])
    vs_full = crowd_model.vss_model(lc3, full_cov = True)

    lc4 = crowd_model.labels_collection(start.turk_data_id[:n], start.rel[:ngold] + (n-ngold)*[None])
    vs_diag = crowd_model.vss_model(lc3, full_cov = False)

    return (gold_dic, mv, tc, vs_full, vs_diag)