datasynth-graph 2.2.0

Graph/network export for synthetic accounting data - supports PyTorch Geometric, Neo4j, and DGL formats
Documentation 
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

#![deny(clippy::unwrap_used)]
//! # synth-graph
//!
//! Graph/network export library for synthetic accounting data.
//!
//! This crate provides:
//! - Graph models for representing accounting data as networks
//! - Builders for creating transaction, approval, and entity graphs
//! - Exporters for PyTorch Geometric, Neo4j, and DGL formats
//! - ML utilities for feature extraction and dataset splitting
//!
//! ## Graph Types
//!
//! - **Transaction Network**: Accounts/entities as nodes, transactions as edges
//! - **Approval Network**: Users as nodes, approvals as edges (for SoD detection)
//! - **Entity Relationship**: Legal entities with ownership edges
//!
//! ## Export Formats
//!
//! - **PyTorch Geometric**: node_features.pt, edge_index.pt, edge_attr.pt
//! - **Neo4j**: CSV files with Cypher import scripts
//! - **DGL**: Compatible format for Deep Graph Library

pub mod builders;
pub mod exporters;
pub mod ml;
pub mod models;

#[cfg(test)]
#[allow(clippy::unwrap_used)]
pub(crate) mod test_helpers;

// Re-export builder types
pub use builders::{
    AccountLinkInput, ApprovalGraphBuilder, ApprovalGraphConfig, BankingGraphBuilder,
    BankingGraphConfig, BuilderInput, ComplianceGraphBuilder, ComplianceGraphConfig,
    ControlLinkInput, CrossReferenceEdgeInput, EntityGraphBuilder, EntityGraphConfig,
    FilingNodeInput, FindingNodeInput, HypergraphBuilder, HypergraphConfig,
    JurisdictionMappingInput, JurisdictionNodeInput, LayerDemand, OwnershipHierarchy,
    OwnershipHierarchyNode, ProcedureNodeInput, SimpleApproval, StandardNodeInput,
    SupersessionEdgeInput, TransactionGraphBuilder, TransactionGraphConfig,
};

// Re-export exporter types
pub use exporters::{
    CommonExportConfig, CommonGraphMetadata, CypherQueryBuilder, DGLExportConfig, DGLExporter,
    DGLMetadata, HypergraphExportConfig, HypergraphExporter, Neo4jExportConfig, Neo4jExporter,
    Neo4jMetadata, PyGExportConfig, PyGExporter, PyGMetadata, RawUnifiedEdge, RawUnifiedHyperedge,
    RawUnifiedNode, RustGraphEdgeMetadata, RustGraphEdgeOutput, RustGraphExportConfig,
    RustGraphExporter, RustGraphMetadata, RustGraphNodeMetadata, RustGraphNodeOutput,
    RustGraphOutputFormat, RustGraphUnifiedExporter, UnifiedExportConfig,
    UnifiedHypergraphMetadata,
};

#[cfg(feature = "rustgraph")]
pub use exporters::RustGraphBulkExport;

// Re-export ML types
pub use ml::*;

// Re-export model types
pub use models::{
    AccountNode, AggregationStrategy, ApprovalEdge, CompanyNode, CrossLayerEdge, EdgeDirection,
    EdgeId, EdgeProperty, EdgeType, Graph, GraphEdge, GraphMetadata, GraphNode, GraphType,
    HeterogeneousGraph, Hyperedge, HyperedgeParticipant, Hypergraph, HypergraphLayer,
    HypergraphMetadata, HypergraphNode, NodeBudget, NodeBudgetReport, NodeBudgetSuggestion, NodeId,
    NodeProperty, NodeType, OwnershipEdge, TransactionEdge, UserNode,
};