I am having the next error when I set the n_stage > 2.
---------------------------------------------------------------------------
ZeroDivisionError Traceback (most recent call last)
<ipython-input-23-9a97b7bcfeb2> in <module>()
----> 1 test_dataset(DATA_NAME="yoochoose", n_stage=4, method="chainRec_uniform", embed_size=16, lbda=0.1)
<ipython-input-22-87d22fb4e825> in test_dataset(DATA_NAME, n_stage, method, embed_size, lbda)
13 target_stage_id=(n_stage-1))
14 myModel.load_samples(training_samples, validation_samples)
---> 15 myModel.train_edgeOpt()
16
17 myModel.evaluation(myData.data_test, myData.user_item_map, topK=10)
<ipython-input-21-021074151cde> in train_edgeOpt(self)
185 _loss_vali += _loss_vali_batch
186 count_sample += len(xu1)
--> 187 _loss_vali /= count_sample
188
189 if _loss_vali <= _loss_vali_min:
ZeroDivisionError: float division by zero
EDIT: Sorry! I think that I missunderstood what n_stage is. I think that this error is because the yoochoose dataset only have 2 stages, clicks and purchases, right?