[HFUT_Student]运行HierCDF模型时IdentifiabilityEvalTPL模板出现问题

[yujiebing]

2024-12-22 20:15:05[ERROR]: Traceback (most recent call last):
File "D:\major assignment\edustudio\quickstart\quickstart.py", line 58, in run_edustudio
traintpl.start()
File "D:\major assignment\edustudio\traintpl\gd_traintpl.py", line 79, in start
metrics = self.one_fold_start(fold_id)
File "D:\major assignment\edustudio\traintpl\general_traintpl.py", line 65, in one_fold_start
metrics = self.inference(self.test_loader)
File "C:\Users\20762\anaconda3\envs\major2\lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
File "D:\major assignment\edustudio\traintpl\general_traintpl.py", line 196, in inference
for evaltpl in self.evaltpls: eval_result.update(evaltpl.eval(**eval_data_dict))
File "D:\major assignment\edustudio\evaltpl\identifiability_evaltpl.py", line 25, in eval
metric_result[metric_name] = self._get_metrics(metric_name)(
File "D:\major assignment\edustudio\evaltpl\identifiability_evaltpl.py", line 101, in get_IDS
return (count / val).item()
ZeroDivisionError: float division by zero

报错出现在IdentifiabilityEvalTPL的get_IDS函数上。可能忽视了interaction matrix中任意两个元素的值都不相同的特殊情况，尽管这个概率很小。是否应该考虑 try-except 块来捕获 ZeroDivisionError 异常。

[kervias]

感谢您的反馈！我们已在代码中进行了修复