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"
