input = diseases_perprocessing(params["input"]) pred = F.softmax(diseases_lstmnn(torch.from_numpy(input).float()), dim=1)[0][1] returnData['prob'] = f'{pred:.4f}'
时间: 2023-06-11 21:04:11 浏览: 71
automatic-diagnosis.rar_leaf diseases_病叶_病斑_病斑特征_颜色纹理形状
5星 · 资源好评率100%
这是一个关于疾病预测的代码段,它通过输入参数params["input"]来进行疾病预处理,并且使用一个基于LSTM的神经网络模型进行预测,返回概率值pred,最后将概率值以字符串形式存储在returnData字典变量中的prob键值对中。
阅读全文