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

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框架下再训练一遍保存才能用

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 10月19日
  • 已采纳回答 10月11日

悬赏问题

  • ¥15 Spark加载本地pipelinemodel报错
  • ¥20 电脑鼠标键盘不好使了
  • ¥15 求糕手!用matlab求解目标规划模型🙏
  • ¥15 画出的分类图不对,求解答一下
  • ¥20 springboot和springcloud版本问题
  • ¥15 ps2手柄控制树莓派小车
  • ¥30 C#:vsto powerpoint的外接程序
  • ¥15 两个同维数组相比,不同位置、出现重复比无意义,而不同位置、不出现重复比,则有意义。把有意义的两个数组放入新的集合MK中。
  • ¥15 写一个可直接调用的函数,将32位有符号数转成另一个无符号的数
  • ¥15 CMAKE+VS2019+QT5.15.2组合进行二次编译