s434765/ium_434765
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

decrease layers
All checks were successful
s434765-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Karolina Oparczyk 2021-05-17 18:48:14 +02:00
parent aa32f7db55
commit 58d72f71c8
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
neural_network.py
View File

@ -19,17 +19,14 @@ y = NormalizeData(y)
model = keras.Sequential([
keras.layers.Dense(512,input_dim = X.shape[1],kernel_initializer='normal', activation='relu'),
keras.layers.Dense(512,kernel_initializer='normal', activation='relu'),
keras.layers.Dense(256,kernel_initializer='normal', activation='relu'),
keras.layers.Dense(256,kernel_initializer='normal', activation='relu'),
keras.layers.Dense(128,kernel_initializer='normal', activation='relu'),
keras.layers.Dense(1,kernel_initializer='normal', activation='linear'),
keras.layers.Dense(32,input_dim = X.shape[1],kernel_initializer='normal', activation='relu'),
keras.layers.Dense(64,kernel_initializer='normal', activation='relu'),
keras.layers.Dense(1,kernel_initializer='normal', activation='relu'),
])
model.compile(loss='mean_absolute_error', optimizer='adam', metrics=['mean_absolute_error'])
model.fit(X, y, epochs=30, validation_split = 0.3)
model.fit(X, y, epochs=15, validation_split = 0.3)
data = pd.read_csv("data_test", sep=',', error_bad_lines=False).dropna()
X_test = data.loc[:,data.columns == "2805317"].astype(int)