Skip to content

Commit 0f70458

Browse files
committed
graph requirement and eval fix
1 parent 25db339 commit 0f70458

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

eval.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from hygdra_forecasting.dataloader.dataloader import StockDataset
44
from hygdra_forecasting.model.eval import validate
55
from torch import cuda, device
6+
from torch.utils.data import DataLoader
7+
import torch.nn as nn
68
from torch import load
79

810
if cuda.is_available():

graph_requirements.txt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
yfinance
2+
pandas
3+
torch
4+
typing
5+
python-multipart
6+
redis
7+
torch_geometric
8+
graph_transformer_pytorch

0 commit comments

Comments
 (0)