Just花语
2021-04-22 22:55
采纳率: 100%
浏览 156
已结题

django中加入pytorch模型

在django的views视图中加载pytorch模型,但是提示没有自己写的model

 

def insert(request):
    if request.method == "POST":
        username = request.POST.get("username")

        text_ints = Init.converts(username)
        new_text_ints = Init.reset_text([text_ints], seq_len=200)  # 注意这里要添加一个[],因为,reset_text处理的二维数据
        text_tensor = torch.from_numpy(new_text_ints)
        load_model = torch.load("E:/Code/Pycharm/Django/NLP/main_other.pkl")

        device = "cuda" if torch.cuda.is_available() else "cpu"

        batch_size = text_tensor.size(0)  # 这里是1
        hs = load_model.init_hidden(batch_size)  # 初始化隐藏状态
        text_tensor = text_tensor.to(device)
        pred, hs = load_model(text_tensor, hs)  # 判断

        pred = torch.round(pred)

        pred = "评论正面" if pred==1 else "评论负面"
        twz = models.message.objects.create(username=username,pred=pred)
        twz.save()
        return redirect('show/')

    return render(request, "insert.html")

 
  • 好问题 提建议
  • 收藏

2条回答 默认 最新

  • Just花语 2021-10-11 11:05
    已采纳

    好像是需要将模型在django框架下再训练一遍保存才能用

    已采纳该答案
    评论
    解决 无用
    打赏 举报
  • 就喔了 2021-07-05 16:59

    请问您的这个问题解决了没有,我也遇见了同样的问题

    评论
    解决 无用
    打赏 举报

相关推荐 更多相似问题