from plotter import Plotter # fix random seed for reproducibility seed = 7 numpy.random.seed(seed) #writing a circle of pixels to the array according to middle coordinates of mouse position plotter=Plotter() 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