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

try InputLayer

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 20:52:45 +02:00
parent ff2edcf654
commit b8961d4f66
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab5/train/train.py
View File

@ -17,7 +17,7 @@ Y_valid = X_valid.pop('stabf')
Y_valid = pd.get_dummies(Y_valid) Y_valid = pd.get_dummies(Y_valid)
model = tensorflow.keras.Sequential([ model = tensorflow.keras.Sequential([
layers.Input(shape=(12,)), layers.InputLayer(shape=(12,)),
layers.Dense(32), layers.Dense(32),
layers.Dense(16), layers.Dense(16),
layers.Dense(2, activation='softmax') layers.Dense(2, activation='softmax')