请教您，怎么设置不同方法分别在CWRU、DIRG、HIT……这些数据集上进行测试

[hujunwei1226]

感谢您提供代码和相关解释！
您解释的create_cache_dataset函数进行信号的选择，那么论文中表2中: 提出的方法Ours 怎么选择在CWRU、DIRG、HIT……这些数据集上进行训练测试？参数设置和各个数据集怎么对应？

[KarlRaphel]

请参考 #17 和 #19

[hujunwei1226]

感谢您，及时的回复！！

[hujunwei1226]

您好，感谢您的代码与及时回复！
1、”您是否可以提供论文的消融实验（w/o CWRU&XJTU）和 (w/o JUST&IMS) 的 FCN 与 Qwen 微调“的代码？
2、另个，“分别在每个子集的测试集上进行测试，也就是Test on Individual Datasets”，这里测试代码是统一用下面这个代码吗？
“我们还额外编写了一个脚本，用于评价测试集中来自XJTU样本的诊断准确率，并输出每一个测试样本的正常信号ID和查询信号ID。”
或者是我理解错了？这里将“ if vib_dataset == 'XJTU':” ,XJTU换成其他数据集如CWRU\MBHM 都不能实现。
不知道能否提供微调BearLLM后，在CWRU、DIRG、HIT、……、MBHM 和 （w/o CWRU&XJTU）、(w/o JUST&IMS) 相关代码及使用说明。