diff --git a/gonito.yaml b/gonito.yaml index 604e67b..80fd96b 100644 --- a/gonito.yaml +++ b/gonito.yaml @@ -1,11 +1,14 @@ description: trigram model tags: - neural-network - - trigram + - ngram params: - epochs: 1 - learning-rate: 0.0001 - vocab_size: 38000 - embed_size: 300 - hidden_size: 128 - batch_size: 800 + - vocab_size = 15000 + - embedding_dim = 200 + - hidden_dim = 500 + - batch_size = 4000 + - lr = 0.0001 + - l_size = 25 + - r_size = 25 + - context_dim = 50 +