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
