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] - Blame information for rev 2

Details | Compare with Previous | View Log

Line No. Rev Author Line
1 2 qaztronic
from tensorflow.keras.models import Sequential
2
from tensorflow.keras.layers import Dense
3
from tensorflow.keras.optimizers import SGD
4
import numpy as np
5
 
6
# -------------------------------------------------------
7
x = np.array([[0,0],[0,1],[1,0],[1,1]])
8
y = np.array([[0],[1],[1],[0]])
9
 
10
# -------------------------------------------------------
11
model = Sequential()
12
model.add(Dense(8, input_dim=x.shape[1], activation='relu'))
13
model.add(Dense(1, activation='sigmoid'))
14
 
15
model.compile(loss='mean_squared_error',
16
              optimizer='adam',
17
              metrics=['binary_accuracy'])
18
 
19
model.fit(x, y, epochs=100000, verbose=0)
20
 
21
print(model.predict(x).round())
22
print(model.predict(x))
23
 
24
# -------------------------------------------------------
25
model.save('xor.h5')
26
model.summary()

powered by: WebSVN 2.1.0

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