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

adjust model

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 21:35:32 +02:00
parent 48f74d8b32
commit 7041cd0703
2 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lab5/eval/requirements.txt
View File

@ -1,10 +1,4 @@
h5py==2.10.0 numpy~=1.19.2
Keras==2.4.3 pandas
Keras-Preprocessing==1.1.2 tensorflow
keras-vis==0.4.1 keras==2.4.3
numpy==1.19.5
pandas==1.1.5
pandas-datareader==0.9.0
pandas-gbq==0.13.3
pandas-profiling==1.4.1
tensorflow==2.4.1

1
lab5/train/train.py
View File

@ -17,7 +17,6 @@ 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.InputLayer(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')