JohanBeytell commited on
Commit
600a01c
·
verified ·
1 Parent(s): a0424ec

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +43 -0
config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "keras",
3
+ "keras_version": "2.11.0",
4
+ "backend": "tensorflow",
5
+ "input_shape": [null, 10],
6
+ "output_shape": [null, 500],
7
+ "description": "A lightweight RNN-based language model for generating fantasy-style weapon and item names, trained on Terraria weapon data.",
8
+ "license": "mit",
9
+ "tags": ["keras", "rnn", "name-generation", "terraria", "text-generation"],
10
+ "author": "Infinitode Pty Ltd",
11
+ "training": {
12
+ "optimizer": "adam",
13
+ "loss": "categorical_crossentropy",
14
+ "metrics": ["accuracy"],
15
+ "epochs": 500,
16
+ "batch_size": 128
17
+ },
18
+ "architecture": {
19
+ "embedding": {
20
+ "input_dim": 500,
21
+ "output_dim": 500
22
+ },
23
+ "recurrent": {
24
+ "type": "SimpleRNN",
25
+ "units": 50
26
+ },
27
+ "output": {
28
+ "type": "Dense",
29
+ "units": 500,
30
+ "activation": "softmax"
31
+ }
32
+ },
33
+ "tokenizer": {
34
+ "type": "SentencePiece",
35
+ "vocab_size": 500,
36
+ "model": "bpe"
37
+ },
38
+ "dataset": {
39
+ "name": "All Terraria Weapons DPS v1.449",
40
+ "source": "https://www.kaggle.com/datasets/acr1209/all-terraria-weapons-dps-v-1449",
41
+ "entries": 395
42
+ }
43
+ }