-
Notifications
You must be signed in to change notification settings - Fork 217
Open
Labels
bugSomething isn't workingSomething isn't working
Description
NGCF produces an index out of bounds error on line:
NeuRec/model/general_recommender/NGCF.py
Line 277 in f8eb375
| weight_initializer([self.weight_size_list[k], self.weight_size_list[k+1]]), name='W_gc_%d' % k) |
IndexError: index 2 is out of bounds for axis 0 with size 2
Properties are set to the following:
[DEFAULT]
data.input.path=neurec
data.input.dataset=ml-100k.rating
data.column.format=UIRT
data.convert.separator='\t'
data.convert.binarize.threshold=0
data.splitter=loo
data.splitterratio=[0.8,0.2]|float
rec.number.thread=20
rec.evaluate.neg=0
recommender=ngcf
topk=[10, 20, 30, 40]|int
gpu_id=0
epochs=500
batch_size=512
embedding_size=16
layer_size=[16,16]|int
learning_rate=0.001
node_dropout_flag=False
adj_type=norm
alg_type=ngcf
loss_function=BPR
learner=adam
reg=0.0
node_dropout_ratio=0.1
mess_dropout_ratio=0.1
#tnormal: truncated_normal_initializer, uniform: random_uniform_initializer,
#normal: random_normal_initializer, xavier_normal, xavier_uniform,
#he_normal, he_uniform. Defualt: tnormal
embed_init_method=xavier_normal
weight_init_method=xavier_normal
stddev=0.01
verbose=1
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working
