import matplotlib.pyplot as plt import numpy as np from sim import Sim from controller import Controller from keras.models import Sequential from keras.optimizers import Adam from keras.layers import Dense, Activation from keras.utils import plot_model from ann_visualizer.visualize import ann_viz import keras.backend as K model = Sequential() model.add(Dense(4, input_dim=1, name="main_1")) model.add(Dense(1)) model.summary() #plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=True) ann_viz(model, title="My first neural network") model.compile(optimizer=Adam(lr=0.1), loss='mae', metrics=['mae']) cntrl = Controller(model) sim_inst = Sim(cntrl) error = sim_inst.run_multiple(1000) print("done")