if __name__ == '__main__': FLAG = '' LEN_ARGV = len(sys.argv) try: DATA_SELECT = sys.argv[1] FLAG = sys.argv[2] except: pass k = 10 if DATA_SELECT == 'i': RAW_DATA = read_informs() ATT_TREES = read_informs_tree() else: RAW_DATA = read_adult() ATT_TREES = read_adult_tree() print '#' * 30 if DATA_SELECT == 'a': print "Adult data" else: print "INFORMS data" print '#' * 30 if FLAG == 'k': get_result_k(ATT_TREES, RAW_DATA) elif FLAG == 'qi': get_result_qi(ATT_TREES, RAW_DATA) elif FLAG == 'data': get_result_dataset(ATT_TREES, RAW_DATA) elif FLAG == 'one': if LEN_ARGV > 3: k = int(sys.argv[3])
_, eval_result = clustering_based_k_anon(att_trees, data, type_alg, k) data = copy.deepcopy(data_back) print("NCP %0.2f" % eval_result[0] + "%") all_ncp.append(round(eval_result[0], 2)) print("Running time %0.2f" % eval_result[1] + "seconds") all_rtime.append(round(eval_result[1], 2)) print("All NCP", all_ncp) print("All Running time", all_rtime) if __name__ == '__main__': print("Using Adult Dataset") DATA = read_adult() ATT_TREES = read_adult_tree() TYPE_ALG = 'oka' FLAG = '' try: FLAG = sys.argv[1] except IndexError: pass if FLAG == 'k': get_result_k(ATT_TREES, DATA, TYPE_ALG) elif FLAG == 'n': get_result_n(ATT_TREES, DATA, TYPE_ALG) elif FLAG == '': if __DEBUG: cProfile.run('get_result_one(ATT_TREES, DATA, TYPE_ALG)') else:
LEN_ARGV = len(sys.argv) try: DATA_SELECT = sys.argv[1] FLAG = sys.argv[2] except IndexError: pass INPUT_L = 5 # read record if DATA_SELECT == 'i': print "INFORMS data" DATA = read_informs() ATTRIBUTE_TREES = read_informs_tree() else: print "Adult data" DATA = read_adult() ATTRIBUTE_TREES = read_adult_tree() if FLAG == 'l': get_result_l(ATTRIBUTE_TREES, DATA) elif FLAG == 'qi': get_result_qi(ATTRIBUTE_TREES, DATA) elif FLAG == 'data': get_result_dataset(ATTRIBUTE_TREES, DATA) elif FLAG == '': get_result_one(ATTRIBUTE_TREES, DATA) else: try: INPUT_L = int(FLAG) get_result_one(ATTRIBUTE_TREES, DATA, INPUT_L) except ValueError: print "Usage: python anonymizer [a | i] [l | qi | data]" print "a: adult dataset, i: INFORMS ataset"