非线性回归Keras实现

用序贯模型实现

'''
用keras实现非线性回归
用Sequential
'''
import keras
import numpy as np
from keras.models import Sequential
from keras.layers import Dense
from keras.optimizers import SGD
import matplotlib.pyplot as plt

x = np.linspace(-0.5, 0.5, 200)
noise = np.random.normal(loc=0, scale=0.01, size=x.shape)
y = np.square(x) + noise

model = Sequential()
model.add(Dense(units=10, input_dim=1, activation='tanh'))
model.add