Downloads · 30 days
0
ARotting/meshgraph-gcn
meshgraph-gcn is a machine learning model from ARotting. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
MeshGraph is a graph convolutional node classifier for detecting compromised assets inside a simulated enterprise network. Nodes carry host telemetry, while edges encode observed communication. Compromise begins at sp…
Downloads · 30 days
0
Access
Public
Updated Jul 30, 2026
Parameters
802
4.1 KB on disk
Likes
0
Public
Click a slice to open those files.
.py10.7 KB · 57%
From the Hugging Face model README
MeshGraph is a graph convolutional node classifier for detecting compromised assets inside a simulated enterprise network. Nodes carry host telemetry, while edges encode observed communication. Compromise begins at sparse seeds and propagates stochastically through the network.
The GCN is compared with a logistic-regression baseline that sees identical host features but cannot use graph structure.
This is a transductive benchmark: the graph and all node features are visible during training, while validation and test labels remain hidden.
uv run python projects/meshgraph-gcn/generate_data.py
uv run python projects/meshgraph-gcn/train.py
The generated graph contains 600 nodes, 1,689 undirected communication edges, six subnets, and a 23.33% compromise rate. The final test contains 120 nodes:
| Model | Parameters | Accuracy | ROC-AUC | Average precision | F1 |
|---|---|---|---|---|---|
| Feature-only logistic regression | 9 fitted coefficients | 78.33% | 0.8362 | 0.6274 | 0.5938 |
| Two-layer GCN | 802 | 81.67% | 0.8564 | 0.6976 | 0.6333 |
Both thresholds were selected independently on the same 120-node validation split. The GCN improved F1 by 3.96 points and average precision by 7.02 points.