diff --git a/code/binary_bpr/script.py b/code/binary_bpr/script.py index a42ffffaa432b66ecc20a3ef90ade1fb47f7b194..f87bf4fd7be4ec040413fd82b36db2eded9548c7 100644 --- a/code/binary_bpr/script.py +++ b/code/binary_bpr/script.py @@ -2,8 +2,8 @@ import os dPath = "../../data/" embDirPath = "../../results/table_2/" datasets = ['assist0910_tkde', 'assist17_tkde', 'algebra','math_1', 'math_2'] -epochs = [1, 75 ,75, 95, 5, 90, 90] -batchSize =[4000, 512, 512,512, 512,512] +epochs = [75 ,75, 95, 5, 90, 90] +batchSize =[ 512, 512,512, 512,512] learningRate = [0.01,0.01,0.01,0.01,0.01] mode = [1,1,1,1,1] for i in range(len(datasets)): diff --git a/data.zip b/data.zip index 4e669c4f2398f54438daa1bbac1216951f67905b..a38901a926a0ba0aa277abe2beb360a38fd9dc3b 100644 Binary files a/data.zip and b/data.zip differ