Commit b6de3606 authored by mihaivanea's avatar mihaivanea
Browse files

trying to fix the way I input the labeled data

parent 22053865
#!/vol/bitbucket/mv1315/urop/venv/bin/python3.5
from keras.layers import Dense, Input
from keras.layers import Dense, Input, Flatten
from keras.models import Model
from generate_arrays_fddb import load_arrays_fddb
......@@ -7,8 +7,8 @@ inputs = Input(shape=(224, 224,3,))
x = Dense(100, activation="relu")(inputs)
x = Dense(100, activation="relu")(x)
predictions = Dense(4, activation="softmax")(x)
x = Flatten()(x)
predictions = Dense(4, activation="relu")(x)
model = Model(inputs=inputs, outputs=predictions)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment