Bering.graphs.CreateData

Bering.graphs.CreateData(bg, batch_size=16, training_ratio=0.8)[source]

Create training and testing data loader

Parameters:

bg (Bering_Graph) – Bering_Graph object
batch_size (int) – Batch size for training
training_ratio (float) – Ratio of training data to the total data

Returns:

: - Bering_Graph.train_loader: Training data loader (torch_geometric.data.DataLoader) - Bering_Graph.test_loader: Testing data loader (torch_geometric.data.DataLoader)