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

:/

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 21:50:50 +02:00
parent 7041cd0703
commit 0ecc2edb39
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lab5/train/train.py
View File

@ -17,6 +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.Dense(32), layers.Dense(32),
layers.Dense(16), layers.Dense(16),
layers.Dense(2, activation='softmax') layers.Dense(2, activation='softmax')