Modelpublic

lucymakeit/dom-node-classifier

sourceHugging Faceapache-2.0updated 5mo agoView on Hugging Face

0likes9downloads

config.json14 linesDownload Raw Back to root

1{2  "model_type": "dom-gat",3  "architecture": "GATv2",4  "in_dim": 618,5  "hidden": 128,6  "n_heads": 4,7  "n_layers": 3,8  "dropout": 0.3,9  "n_classes": 14,10  "add_reverse_edges": true,11  "add_sibling_edges": true,12  "text_encoder": "sentence-transformers/all-MiniLM-L6-v2"13}14