s434784/ium_434784
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

.
Some checks failed
s434784-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Maciej Sobkowiak 2021-05-16 16:55:33 +02:00
parent eb520d69b7
commit 5687a9cc98
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
training.py
View File

@ -64,8 +64,8 @@ model.compile(
# Train model # Train model
history = model.fit( history = model.fit(
X_train, y_train, X_train, y_train,
batch_size=sys.argv[0], batch_size=int(sys.argv[0]),
epochs=sys.argv[1], epochs=int(sys.argv[1]),
validation_split=0.2) validation_split=0.2)
test_results = {} test_results = {}