s470607/ium_470607
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

model adds

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 22:06:36 +02:00
parent b676105f9d
commit 8324f60cd4
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lab5/train/train.py
View File

@ -16,12 +16,18 @@ Y_test = pd.get_dummies(Y_test)
Y_valid = X_valid.pop('stabf')
Y_valid = pd.get_dummies(Y_valid)
model = tensorflow.keras.Sequential([
layers.Input(shape=(12,)),
layers.Dense(32),
layers.Dense(16),
layers.Dense(2, activation='softmax')
])
# model = tensorflow.keras.Sequential([
# layers.Input(shape=(12,)),
# layers.Dense(32),
# layers.Dense(16),
# layers.Dense(2, activation='softmax')
# ])
model = tensorflow.keras.Sequential()
model.add(layers.Input(shape=(12,)))
model.add(layers.Dense(32))
model.add(layers.Dense(16))
model.add(layers.Dense(2, activation='softmax'))
model.compile(
loss=tensorflow.keras.losses.BinaryCrossentropy(),