Skip to content

Commit ccb87ca

Browse files
committed
Add info about 250 graphs
1 parent 57141aa commit ccb87ca

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

GraphNeuralNetworks/docs/src/tutorials/temporal_graph_classification.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ dataset TemporalBrains:
3939
````
4040

4141
After loading the dataset from the MLDatasets.jl package, we see that there are 1000 graphs and we need to convert them to the `TemporalSnapshotsGNNGraph` format.
42-
So we create a function called `data_loader` that implements the latter and splits the dataset into the training set that will be used to train the model and the test set that will be used to test the performance of the model.
42+
So we create a function called `data_loader` that implements the latter and splits the dataset into the training set that will be used to train the model and the test set that will be used to test the performance of the model. Due to computational costs, we use only 250 out of the original 1000 graphs, 200 for training and 50 for testing.
4343

4444
````julia
4545
function data_loader(brain_dataset)

GraphNeuralNetworks/docs/src_tutorials/introductory_tutorials/temporal_graph_classification.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Random.seed!(17); # for reproducibility
3232
brain_dataset = TemporalBrains()
3333

3434
# After loading the dataset from the MLDatasets.jl package, we see that there are 1000 graphs and we need to convert them to the `TemporalSnapshotsGNNGraph` format.
35-
# So we create a function called `data_loader` that implements the latter and splits the dataset into the training set that will be used to train the model and the test set that will be used to test the performance of the model.
35+
# So we create a function called `data_loader` that implements the latter and splits the dataset into the training set that will be used to train the model and the test set that will be used to test the performance of the model. Due to computational costs, we use only 250 out of the original 1000 graphs, 200 for training and 50 for testing.
3636

3737

3838
function data_loader(brain_dataset)

0 commit comments

Comments
 (0)