gonito.yaml

@@ -1,14 +1,11 @@
 description: trigram model
 tags:
   - neural-network
-  - ngram
+  - trigram
 params:
-  - vocab_size = 15000
-  - embedding_dim = 200
-  - hidden_dim = 500
-  - batch_size = 4000
-  - lr = 0.0001
-  - l_size = 25
-  - r_size = 25
-  - context_dim = 50
-  
+  epochs: 1
+  learning-rate: 0.0001
+  vocab_size: 38000
+  embed_size: 300
+  hidden_size: 128
+  batch_size: 800