From a9e14b91484db6b8cb15be3faf1bec35442e9ada Mon Sep 17 00:00:00 2001 From: Siddharth Rao Date: Sat, 22 Jun 2024 02:09:53 -0700 Subject: [PATCH] Categorical Model Architecture --- src/training/_build_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/training/_build_model.py b/src/training/_build_model.py index da180c3..54df24e 100644 --- a/src/training/_build_model.py +++ b/src/training/_build_model.py @@ -48,7 +48,7 @@ def build_model(): Dense(64, activation='relu'), Dropout(0.5), - Dense(1, activation='linear', name='value'), + Dense(3, activation='linear', name='value'), ]) model.compile(