for i, st in enumerate(str_list): st.set_material(str_colours[i, 0]) st.scale(0.5 + outlayer[i], 1.0, 1.0) ready_flag = False k = mykeys.read() if k >-1: if k == 27: break if k >= ord('1') and k <= ord('5'): # do training trained = True i = k - ord('1') targets = np.zeros((5,), dtype=np.float) targets[i] = 1.0 presses[i] += 1 print('{} {},{},{},{},{}'.format(mlp.back_propagate(targets, learning_rate=0.05), *presses)) ''' NB uncomment below with care; this will generate a lot of images very quickly! Also you obviously need a viable path rather than the one below which will only match the USB drive I happen to be using. ''' #pi3d.screenshot("/media/pi/701E-64FC/tmp/scr_cap/fr{:05d}.jpg".format(fr)) #fr += 1 mykeys.close() DISPLAY.destroy() if trained: np.save('wi_file.npy', mlp.wi) np.save('wo_file.npy', mlp.wo) try: with open('presses.txt','r') as f: