ohyeon5 / pytorch-graphnet Goto Github PK

I don't know if you are Korean or not, so I write in English and Korean.

Hello. I'm a graduate student studying about deep learning.
Nowadays I'm studying graphnet developed by DeepMind (https://github.com/deepmind/graph_nets).

I'm interested about 'Shortest Path demo' so I'll talk about this.
Well, in my case, training is perfect so far.
When it comes to apply test data, however, result is very bad.
Model cannot find the shortest path.
Can you help me about this problem?
Can you tell me about the form of test data?
Really appreciate if you tell me about how to apply test data on trained model.

Thank you.

안녕하세요 최근 graphnet을 공부하고 있는 대학원생입니다.
DeepMind에서 개발한 graphnet의 'Shortest Path demo'를 공부하다 궁금한점이 있어 글을 남깁니다.
저는 서울시 도로 노드 엣지 데이터를 이용하여 트레이닝을 시켰습니다.
트레이닝이 잘 되는 것을 확인 후 테스트 데이터를 트레이닝 된 모델에 적용하였습니다.
테스트 데이터는 시작 노드와 도착 노드의 정보만을 True값을 할당하였고 최단거리에 해당하는 중간 노드와 엣지의 정보는 False로 할당하여 모델에 넣어주었습니다.
그랬더니 아무런 결과를 얻지 못하였습니다.
혹시 제가 어떤 점을 잘못했는지 알려주실수 있으신가요?
테스트 데이터를 어떤식으로 구성해야 하는지에 대해 알려주실수 있으신가요?
자료가 너무 없어 이렇게 글을 남깁니다.
감사합니다.