File size: 934 Bytes
484e3bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
"""
Machine Learning enhancers for GeoBotv1
Optional but powerful additions once the causal backbone is built.
Critical principle: These help discover new relationships but must not replace causality.
"""
from .risk_scoring import RiskScorer
from .feature_discovery import FeatureDiscovery
from .embedding import GeopoliticalEmbedding
# GNN imports are optional (require PyTorch)
try:
from .graph_neural_networks import (
CausalGNN,
GeopoliticalNetworkGNN,
AttentionGNN,
MessagePassingCausalGNN,
GNNTrainer,
NetworkToGraph
)
_has_gnn = True
except ImportError:
_has_gnn = False
__all__ = [
"RiskScorer",
"FeatureDiscovery",
"GeopoliticalEmbedding",
]
if _has_gnn:
__all__.extend([
"CausalGNN",
"GeopoliticalNetworkGNN",
"AttentionGNN",
"MessagePassingCausalGNN",
"GNNTrainer",
"NetworkToGraph",
])
|