NN=Neural_Network() if input("Wanna train model? y/N ") =='y': layers = [800,400] NN.initialize(layers) NN.train(20) if input("Wanna save model? Y/n ") !='n' : NN.save_model() elif input("Wanna load from file? Y/n ")!='n' : NN.load_model() else : print("Exit") exit() # Fit the model # Final evaluation of the model if input("Wanna test model? y/N ") =='y': NN.test() #space for insertion of numpy array from user: while input("Wanna input some stuff? Y/n ")!='n': NN.predict(plotter.input_stuff()) plotter.flush()