Skip to content

Commit

Permalink
Add manual seed to introduction examples
Browse files Browse the repository at this point in the history
Manual seed and updated MSE scores.
  • Loading branch information
boykovdn committed Jun 18, 2023
1 parent 1c081e3 commit a586da8
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions docs/source/notes/introduction.rst
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,7 @@ In the next steps we will define the **recurrent graph neural network** architec
.. code-block:: python
import torch
torch.manual_seed(1)
import torch.nn.functional as F
from torch_geometric_temporal.nn.recurrent import DCRNN
Expand Down Expand Up @@ -194,7 +195,7 @@ Using the holdout we will evaluate the performance of the trained recurrent grap
cost = cost / (time+1)
cost = cost.item()
print("MSE: {:.4f}".format(cost))
>>> MSE: 0.7302
>>> MSE: 0.7418
Web Traffic Prediction
----------------------
Expand All @@ -218,6 +219,7 @@ In the next steps we will define the **recurrent graph neural network** architec
.. code-block:: python
import torch
torch.manual_seed(1)
import torch.nn.functional as F
from torch_geometric_temporal.nn.recurrent import GConvGRU
Expand Down Expand Up @@ -265,4 +267,4 @@ Using the holdout traffic data we will evaluate the performance of the trained r
cost = cost / (time+1)
cost = cost.item()
print("MSE: {:.4f}".format(cost))
>>> MSE: 0.5114
>>> MSE: 0.5264

0 comments on commit a586da8

Please sign in to comment.