s434742/ium_434742
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

minor fix
Some checks failed
s434742-training/pipeline/head There was a failure building this commit
Details
s434742-evaluation/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
patrycjalazna 2021-05-14 20:58:03 +02:00
parent e42afc40ed
commit 4fc45ed8ef
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sacred-training.py
View File

@ -47,10 +47,6 @@ def prepare_model():
# kompilacja # kompilacja
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
# model fit
epochs = int(sys.argv[1])
batch_size = int(sys.argv[2])
# trenowanie modelu # trenowanie modelu
model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size, validation_data=(X_test, y_test)) model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size, validation_data=(X_test, y_test))