OpenCores
URL https://opencores.org/ocsvn/keras_to_fpga/keras_to_fpga/trunk

Subversion Repositories keras_to_fpga

[/] [keras_to_fpga/] [trunk/] [src/] [xor/] [xor_model.py] - Rev 2

Compare with Previous | Blame | View Log

from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
from tensorflow.keras.optimizers import SGD
import numpy as np 
 
# -------------------------------------------------------
x = np.array([[0,0],[0,1],[1,0],[1,1]])
y = np.array([[0],[1],[1],[0]])
 
# -------------------------------------------------------
model = Sequential()
model.add(Dense(8, input_dim=x.shape[1], activation='relu'))
model.add(Dense(1, activation='sigmoid'))
 
model.compile(loss='mean_squared_error',
              optimizer='adam',
              metrics=['binary_accuracy'])
 
model.fit(x, y, epochs=100000, verbose=0)
 
print(model.predict(x).round())
print(model.predict(x))
 
# -------------------------------------------------------
model.save('xor.h5')
model.summary()
 

Compare with Previous | Blame | View Log

powered by: WebSVN 2.1.0

© copyright 1999-2024 OpenCores.org, equivalent to Oliscience, all rights reserved. OpenCores®, registered trademark.