如何在Django中加载经过训练的模型

2024-06-01 00:49:02 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在django项目中工作,我必须使用Doc2Vec模型根据用户输入预测大多数类似的文章。我在数据库中的文章的帮助下训练了一个模型,当我使用python文件.py测试该模型时,右键单击该文件并从上下文菜单中选择run,它将正常工作。现在的问题是,我正在将工作代码移动到django函数中,以便根据用户给定的抽象文本加载模型和预测文章,但是我得到了FileNotFoundError
我已经搜索了如何在django中加载模型,似乎这种方法已经可以了。以下是一个完全例外:

FileNotFoundError at /searchresult
[Errno 2] No such file or directory: 'd2vmodel.model'
Request Method: GET
Request URL: http://127.0.0.1:8000/searchresult
Django Version: 3.1.5
Exception Type: FileNotFoundError
Exception Value:
[Errno 2] No such file or directory: 'd2vmodel.model'
Exception Location: C:\Users\INZIMAM_TARIQ\AppData\Roaming\Python\Python37\site-packages\smart_open\smart_open_lib.py, line 346, in _shortcut_open
Python Executable: C:\Program Files\Python37\python.exe
Python Version: 3.7.9
Python Path:
['D:\Web Work\doc2vec final submission',
'C:\Program Files\Python37\python37.zip',
'C:\Program Files\Python37\DLLs',
'C:\Program Files\Python37\lib',
'C:\Program Files\Python37',
'C:\Users\INZIMAM_TARIQ\AppData\Roaming\Python\Python37\site-packages',
'C:\Program Files\Python37\lib\site-packages']
Server time: Mon, 24 May 2021 12:44:47 +0000
D:\Web Work\doc2vec final submission\App\views.py, line 171, in searchresult
model = Doc2Vec.load("d2vmodel.model")

这是我的django函数,我在其中加载Doc2Vec模型

def searchresult(request):
    articles = []

    all_articles = Article.objects.all()

    model = Doc2Vec.load("d2vmodel.model")

    def output_sentences(most_similr):
        print('\n')
        for label, index in [('MOST Similar', 0), ('Second-Most Similar', 1), ('Third Most Similar', 3),
                             ('Fourth Most Similar', 4)]:
            print(u'%s %s: %s\n' % (label, most_similar[index][1], all_articles[int(most_similar[index][0])]))
            articles.append(all_articles[int(most_similar[index][0])])
        print('=====================\n')

    # to find the vector of a document which is not in training data
    seed_text = 'Test from frontend dashboard'
    tokens = seed_text.split()
    vector = model.infer_vector(tokens)
    most_similar = model.dv.most_similar([vector])

    output_sentences(most_similar)

    article = Article.objects.filter(abstract=articles)
    art_dict = {
        'articles': article
    }

    return render(request, 'searchresult.html', art_dict)

我在同一目录/文件夹中有d2vmodel.model文件和views.pyenter image description here

请让我知道在django加载doc2vec模型需要做什么。
我正在使用Python3、PyCharm IDE,并使用pip命令安装了libs

多谢各位;问候


Tags: djangoinpy模型mostmodelfilesall
1条回答
网友
1楼 · 发布于 2024-06-01 00:49:02

将模型从应用程序移动到项目的根目录,我认为这是“doc2vec最终提交”

或者在“doc2vec最终提交”中创建一个名为“模型”的文件夹

改变这个

model = Doc2Vec.load("d2vmodel.model")

model = Doc2Vec.load("models/d2vmodel.model")

相关问题 更多 >