Tomorrowdawn / EvolveGCN

Semester project for social computing of USTC.
0 stars 3 forks source link

实现generator.py #5

Closed RuibaiXu closed 2 months ago

RuibaiXu commented 2 months ago

先PR一下,如果可能的话可以帮忙进行下 code review(略读下代码有什么问题) 需要进行测试, 感觉会有很多BUG

Tomorrowdawn commented 2 months ago

next方法是不是要重置index

RuibaiXu commented 2 months ago

generator实例化后跑train_model没问题,DEBUG完成,申请合并,合并前记得把 train_model.py 里加的测试代码去了