def main_(): if len(argv) < 2: print 'The input format is\nmain train/test [tcp/b] [filename]' return 0 t=argv[1] mode='tcp' data=[] if len(argv) == 4 : mode=argv[2] if mode == 'tcp': getdata = getdata_tcp else : getdata = getdata_b if t == 'test': if len(argv) == 2: test() else: fname=argv[3] data=getdata_file(fname) data=filter_xyz(data) data14=cluster(data,14); simulate(data14); elif t == 'train': train() else: while(True): data=getdata() data=filter_xyz(data) data14=cluster(data,14) simulate(data14)
def main_(): if len(argv) < 2: print 'The input format is\nmain train/test [tcp/b] [filename]' return 0 t = argv[1] mode = 'tcp' data = [] if len(argv) == 4: mode = argv[2] if mode == 'tcp': getdata = getdata_tcp else: getdata = getdata_b if t == 'test': if len(argv) == 2: test() else: fname = argv[3] data = getdata_file(fname) data = filter_xyz(data) data14 = cluster(data, 14) simulate(data14) elif t == 'train': train() else: while (True): data = getdata() data = filter_xyz(data) data14 = cluster(data, 14) simulate(data14)
def get_train_data(): from os import listdir,chdir,remove,rename from filtering import filter_xyz from clustering import cluster table=shape_dict() in_put=[] target=[] chdir('./Shapes') for i in listdir('.'): try: data=getdata_file(i) data=filter_xyz(data) data14=cluster(data,14); in_put.append(data14) target.append(table[i[0]]) except: print '\t\t\tERROR in file ',i,'removing ',i remove(i) pass chdir('..') return in_put,target