Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- x_train = np.array([cv2.imread("1small.jpg")])
- y_train = np.array([14.1, 13.5, 16.3, 14.1])
- x_test = np.array([cv2.imread("0small.jpg")])
- y_test = np.array([11.8, 10.8, 17.0, 16.0]) # fake data just to test
- model = Sequential()
- model.add(Dense(1,32,32,3))
- model.add(Activation('tanh'))
- model.add(Dense(1))
- model.compile(loss='mean_absolute_error', optimizer='rmsprop')
- model.fit(x_train, y_train, nb_epoch=1, batch_size=1)
- prediction = model.predict(x_test)
- print prediction
Add Comment
Please, Sign In to add comment