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

.
All checks were successful
s434784-training/pipeline/head This commit looks good
Details

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

6
training.py
View File

@ -14,6 +14,8 @@ from tensorflow import keras
from tensorflow.keras import layers from tensorflow.keras import layers
from tensorflow.keras.layers.experimental import preprocessing from tensorflow.keras.layers.experimental import preprocessing
EPOCHS = int(sys.argv[1])
BATCH_SIZE = int(sys.argv[2])
age = {"5-14 years": 0, "15-24 years": 1, "25-34 years": 2, age = {"5-14 years": 0, "15-24 years": 1, "25-34 years": 2,
"35-54 years": 3, "55-74 years": 4, "75+ years": 5} "35-54 years": 3, "55-74 years": 4, "75+ years": 5}
@ -64,8 +66,8 @@ model.compile(
# Train model # Train model
history = model.fit( history = model.fit(
X_train, y_train, X_train, y_train,
batch_size=int(sys.argv[0]), batch_size=BATCH_SIZE,
epochs=int(sys.argv[1]), epochs=EPOCHS,
validation_split=0.2) validation_split=0.2)
test_results = {} test_results = {}