Apppublic

rudrani-rane/ATIS

sourceHugging Facemitupdated 7mo agoView on Hugging Face

0likes

test_model.py13 linesDownload Raw Back to models

1#Sanity Check 
2import torch
3from src.graph.graph_builder import build_graph
4from src.models.gnn_model import ATISGNN
5
6graph = build_graph()
7
8model = ATISGNN(in_channels=graph.x.shape[1])
9
10mu, sigma = model(graph.x, graph.edge_index)
11
12print("Mu shape:", mu.shape)
13print("Sigma shape:", sigma.shape)