keras-test

test.py

from keras.layers import Lambda
from keras.models import Sequential
from keras.optimizers import RMSprop

# declares toy model that just multiplies input by 2

def mult(x):
    return x * 2

# declare sequential layer
model = Sequential()
lam = Lambda(mult,input_shape=(1,), name="out")

model.add(lam)

# optimizer
optim = RMSprop()

# compile the model
model.compile(optimizer=optim, loss='mean_squared_error')

# training data
X = np.zeros((1,1))
y = np.zeros((1,1))

# train
model.fit(X,y,epochs=1)
    
# run new input against model
#inData = np.zeros((1))
#inData[0] = 8
#print(inData)
#print(model.predict(inData))