feat: add osm data

2025-09-21 16:23:44 +02:00
parent 63e23d6600
commit c1b502c878
39 changed files with 955 additions and 338 deletions
@@ -36,9 +36,9 @@ class TransformerModel(NeuralNetworkModel):

        # Add positional encoding
        positions = tf.range(start=0, limit=params.get("max_len", 8), delta=1)
-        pos_embedding = Embedding(input_dim=params.get("max_len", 8), output_dim=params.get("embedding_dim", 64))(
-            positions
-        )
+        pos_embedding = Embedding(
+            input_dim=params.get("max_len", 8), output_dim=params.get("embedding_dim", 64)
+        )(positions)
        x = x + pos_embedding

        x = self._transformer_encoder(x, params)