文件名称:beacon
文件大小:14KB
文件格式:ZIP
更新时间:2024-04-02 06:11:21
Python
这是“购物”文件(IJCAI'19)中的Beacon复制项目。 输入格式:训练/验证/测试集具有相同的格式 对于每个购物篮序列,购物篮{b_i}以'|'分隔,例如b_1 | b_2 | b_3 | ... | b_n 对于每个购物篮b_i,项目{v_j}之间用空格''隔开,例如v_1 v_2 v_3 ... v_m 训练方法: 步骤1:使用cmatrix_generator.py生成预先计算的相关矩阵C。 “ nbhop”参数用于生成N阶相关矩阵 默认输出目录是“ data_dir / adj_matrix” 第2步:使用main_gpu.py“训练Beacon模型。 支持3种模式:train_mode,prediction_mode,tune_mode 预测文件的格式如下:Target:gt_basket | item_candidate_1:score_1 | item
【文件预览】:
beacon-master
----README.md(1KB)
----main_gpu.py(8KB)
----models.py(10KB)
----cmatrix_generator.py(2KB)
----utils.py(7KB)
----.idea()
--------.gitignore(214B)
--------misc.xml(196B)
--------beacon.iml(284B)
--------modules.xml(264B)
--------inspectionProfiles()
----procedure.py(8KB)
----layers.py(2KB)