Categorical Model Architecture

silverlightning926 · Jun 22, 2024 · a9e14b9 · a9e14b9
1 parent b23c0a1
commit a9e14b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(