You can deploy this model and start earning money today!

alirezaaminzadeh/
threatgraph-gnn-embedder

Name: threatgraph-gnn-embedder
Availability: InStock

joblib

Views

No views yet

ThreatGraph GNN Embedder

Graph Neural Network embedder and baseline classifiers for threat intelligence IOC graphs.

Models

Artifact	Algorithm	Task
`graphsage.pt`	GraphSAGE	Node embedding
`gat.pt`	Graph Attention Network	Node embedding
`gcn.pt`	Graph Convolutional Network	Node embedding
`node_classifier.pt`	MLP head	Node classification
`link_predictor.pt`	MLP scorer	Link prediction
`node2vec_embeddings.json`	Node2Vec	Baseline embeddings
`logistic_regression.joblib`	LR on Node2Vec	Baseline classifier
`random_forest.joblib`	RF on Node2Vec	Baseline classifier

Usage

python

from threatgraph.pipeline import ThreatGraphPipeline
pipeline = ThreatGraphPipeline(data_dir="data", model_dir=".")
result = pipeline.analyze("evil.example.com")
print(result.risk_score, result.malicious_probability)

Evaluation

See train_metrics.json for reproducible benchmark results.

License

MIT — Aria AI Engineering Team.

alirezaaminzadeh/threatgraph-gnn-embedder