sylv/fyp
1
0
Fork 0
Code Issues 29 Pull Requests Packages Projects 1 Releases Wiki Activity

feat: closes #22

Browse Source
This commit is contained in:
Andre Henriques
2023-09-29 13:27:43 +01:00
parent dfa62118de
commit bc948d4796
4 changed files with 46 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
views/py/python_model_template.py
View File

@@ -11,8 +11,9 @@ dataset = keras.utils.image_dataset_from_directory(
"{{ .DataDir }}",
color_mode="rgb",
validation_split=0.2,
label_mode='int',
label_mode='categorical',
seed=seed,
shuffle=True,
subset="training",
image_size=({{ .Size }}),
batch_size=batch_size)
@@ -21,8 +22,9 @@ dataset_validation = keras.utils.image_dataset_from_directory(
"{{ .DataDir }}",
color_mode="rgb",
validation_split=0.2,
label_mode='int',
label_mode='categorical',
seed=seed,
shuffle=True,
subset="validation",
image_size=({{ .Size }}),
batch_size=batch_size)
@@ -42,11 +44,11 @@ model = keras.Sequential([
])
model.compile(
loss=losses.SparseCategoricalCrossentropy(),
loss=losses.CategoricalCrossentropy(),
optimizer=tf.keras.optimizers.Adam(),
metrics=['accuracy'])
his = model.fit(dataset, validation_data= dataset_validation, epochs=70)
his = model.fit(dataset, validation_data= dataset_validation, epochs=50)
acc = his.history["accuracy"]