Update run_VGCL.py

yimutianyang · Jun 16, 2023 · 73c0c97 · 73c0c97
1 parent 377a867
commit 73c0c97
Showing 1 changed file with 1 addition and 12 deletions.
diff --git a/run_VGCL.py b/run_VGCL.py
@@ -145,21 +145,10 @@ def run_kmeans(x, num_cluster):
 
     #***********************************  start evaluate testdata   ********************************#
     writer.close()
-    u1, u2, u3, u4, u5 = rec_data._user_group()
     saver.restore(sess, model_save_path + best_ckpt)
     user_matrix, item_matrix = sess.run([rec_model.user_emb, rec_model.item_emb])
-    hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata, [20], user_matrix, item_matrix, u1)
-    log.write('Group1, topk:{:d}, recall:{:.4f}, ndcg{:.4f}\n'.format(20, recall[20], ndcg[20]))
-    hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata, [20], user_matrix, item_matrix, u2)
-    log.write('Group2, topk:{:d}, recall:{:.4f}, ndcg{:.4f}\n'.format(20, recall[20], ndcg[20]))
-    hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata, [20], user_matrix, item_matrix, u3)
-    log.write('Group3, topk:{:d}, recall:{:.4f}, ndcg{:.4f}\n'.format(20, recall[20], ndcg[20]))
-    hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata, [20], user_matrix, item_matrix, u4)
-    log.write('Group4, topk:{:d}, recall:{:.4f}, ndcg{:.4f}\n'.format(20, recall[20], ndcg[20]))
-    hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata, [20], user_matrix, item_matrix, u5)
-    log.write('Group5, topk:{:d}, recall:{:.4f}, ndcg{:.4f}\n'.format(20, recall[20], ndcg[20]))
     hr, recall, ndcg = num_faiss_evaluate(rec_data.testdata, rec_data.traindata,
                                           [5,10,20,30,40,50,60,70,80,90,100], user_matrix, item_matrix, rec_data.testdata.keys())
     for key in ndcg.keys():
         log.write('Topk:{:3d}, HR:{:.4f}, Recall:{:.4f}, NDCG:{:.4f}\n'.format(key, hr[key], recall[key], ndcg[key]))
-    log.close()
+    log.close()