RippleNet-PyTorch icon indicating copy to clipboard operation
RippleNet-PyTorch copied to clipboard

verified publisher icon qibinc
→

Metadata

A PyTorch implementation of RippleNet

Results 2 RippleNet-PyTorch issues
Sort by recently updated
recently updated
newest added

pytorch与TF结果相差太大

2 comment

你可以试一下将n_hop设置为4，测试结果AUC在0.5左右，但TF版本的可以达到0.91多

PJinBeiJing avatar PJinBeiJing

内存泄漏

在增加hop和增大set后，训练多了会使内存泄漏，64G内存都不好使

PJinBeiJing avatar PJinBeiJing

← Metadata

111
Stars
44
Forks
Watchers

Owner

verified publisher icon qibinc

Metadata

A PyTorch implementation of RippleNet

Back