执行快速 API 代码时出现此错误fastapi.exceptions.FastAPIError: Invalid args for response field! Hint: check that <class 'main.SoluteSolvent'> is a valid pydantic field type
class SoluteSolvent():
solvent: Text
solute: Text
response = {}
@app.get('/predict')
def predict(sol:SoluteSolvent):
data = sol.dict()
solute = data['solute']
solvent = data['solvent']
results = predictions(solute, solvent)
print(results)
response["predictions"] = results[0].item()
response["interaction_map"] = (results[1].detach().numpy()).tolist()
return {'result':response}
Run Code Online (Sandbox Code Playgroud)
我实际上试图在快速 API 中复制我为 Flask 中的预测编写的代码。
response = {}
@app.route('/predict', methods=["POST", "GET"])
def predict():
if request.method=='POST':
solute = request.form["solute"]
solvent = request.form["solvent"]
else:
solute = request.args.get("solute")
solvent …Run Code Online (Sandbox Code Playgroud) 我有一个张量列表,我想将其转换为浮点,我该怎么做。我尝试过使用.item但它不起作用。我正进入(状态ValueError: only one element tensors can be converted to Python scalars。
tensor([[12.1834, 4.9616, 7.7913],
[ 8.9394, 8.5784, 9.3691],
[ 9.4475, 8.9766, 9.8418],
[11.8210, 6.0852, 8.2168],
[ 8.2885, 6.2607, 9.8877]], grad_fn=<CloneBackward0>)
Run Code Online (Sandbox Code Playgroud)
我正在使用 Flask 进行推理,得到这个结果。他们有什么方法可以将此张量转换为浮点数,因为我想使用此结果在反应应用程序中显示
{
result: {
predictions: "tensor([[-3.4333]], grad_fn=<AddmmBackward>)"
}
}
Run Code Online (Sandbox Code Playgroud)