fix: normalize hyper params

This commit is contained in:
2025-09-21 13:10:07 +02:00
parent 83d21c640b
commit 63e23d6600
8 changed files with 26 additions and 19 deletions
+1 -2
View File
@@ -13,7 +13,7 @@ from research.neural_network_model import NeuralNetworkModel
class BiGRUModel(NeuralNetworkModel):
"""Bidirectional GRU model for name classification"""
def build_model_with_vocab(self, vocab_size: int, max_len: int = 6, **kwargs) -> Any:
def build_model_with_vocab(self, vocab_size: int, **kwargs) -> Any:
params = kwargs
model = Sequential(
[
@@ -22,7 +22,6 @@ class BiGRUModel(NeuralNetworkModel):
Embedding(
input_dim=vocab_size,
output_dim=params.get("embedding_dim", 64),
input_length=max_len,
mask_zero=True,
),
# First recurrent block returns full sequences to allow stacking.