ingra14m / graduationproject Goto Github PK

学长，在我的设备上，main函数跑了一段事件后出现了以下问题，希望学长可以帮助。

学长，该代码有实现DrugA-DrugB DDI预测的模块么，就是假设我输入A与B来预测DDI的概率的模块。不胜感激。

你好，请问该项目的运行效果是什么样的呢
我在运行时会出现以下错误，请问可以帮忙解答下吗 不胜感激
D:\GP\GraduationProject-main\src\utils\FocalLoss.py:43: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
P = F.softmax(inputs)
Traceback (most recent call last):
File "D:/GP/GraduationProject-main/src/main.py", line 183, in
train(model=model, optimizer=optimizer, graph=graph, output=args.output)
File "D:/GP/GraduationProject-main/src/main.py", line 61, in train
loss = loss_function(pred[graph.edata['train_mask']], edata_label[graph.edata['train_mask']])
File "C:\Users\BinQiang Zhang.conda\envs\Sum\lib\site-packages\torch\nn\modules\module.py", line 1110, in call_impl
return forward_call(*input, **kwargs)
File "D:\GP\GraduationProject-main\src\utils\FocalLoss.py", line 48, in forward
class_mask.scatter(1, ids.data, 1.)
RuntimeError: scatter(): Expected dtype int64 for index