Skip to content

Commit

Permalink
Changed Metric To Categorical Cross Entropy
Browse files Browse the repository at this point in the history
  • Loading branch information
silverlightning926 committed Jun 21, 2024
1 parent f0080c5 commit 02c363f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/training/_build_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from keras.api.layers import Input, Conv2D, BatchNormalization, LeakyReLU, GlobalAveragePooling2D, Dense, Dropout, TimeDistributed, GRU, Add, LayerNormalization, MultiHeadAttention
from keras.api.optimizers import Adam
from keras.api.regularizers import l2
from keras.api.metrics import Accuracy, BinaryCrossentropy, AUC, Precision, Recall, MeanSquaredError
from keras.api.metrics import Accuracy, CategoricalCrossentropy, AUC, Precision, Recall, MeanSquaredError
from src.utils.path_utils import find_project_directory

from src.training._load_dataset import MAX_MOVES, BATCH_SIZE
Expand Down Expand Up @@ -67,7 +67,7 @@ def build_model():
model.compile(
optimizer=Adam(learning_rate=1e-4),
loss='category_crossentropy',
metrics=[Accuracy(), BinaryCrossentropy(), AUC(), Precision(),
metrics=[Accuracy(), CategoricalCrossentropy(), AUC(), Precision(),
Recall(), MeanSquaredError()]
)

Expand Down

0 comments on commit 02c363f

Please sign in to comment.