Exemplo n.º 1
0
        else:
            dict_repair[repair_key][0] += entry[-1]
    else:
       error_count += 1

data = []
for value in dict_repair.values():
    data.append([ele for ele in value])

X =  np.array(data)
X = X[:,[0,1,2,3,6]]

fac1 = u.strings_to_classes(X[:,2])
fac2 = u.strings_to_classes(X[:,3])

t1 = u.create_t_matrix(fac1)
t2 = u.create_t_matrix(fac2)

X = np.hstack([np.float32(X[:,[0,1,4]]),t1,t2])
print X.shape

np.save('/home/tim/Downloads/repair/train.npy',X)
print 'Saved!'

#TODO: use util to create categories
#print(t1.ix[0:5,:])
#print(t2.ix[0:5,:])

#print data