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

时间:2021-05-24 13:00:43

标签: python django exception doc2vec

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

<块引用>

FileNotFoundError at /searchresult
[Errno 2] 没有那个文件或目录:'d2vmodel.model'
请求方式:GET
请求网址:http://127.0.0.1:8000/searchresult
Django 版本:3.1.5
异常类型:FileNotFoundError
异常值:
[Errno 2] 没有那个文件或目录:'d2vmodel.model'
异常位置:C:\Users\INZIMAM_TARIQ\AppData\Roaming\Python\Python37\site-packages\smart_open\smart_open_lib.py, line 346, in _shortcut_open
Python 可执行文件:C:\Program Files\Python37\python.exe
Python 版本:3.7.9
Python路径:
['D:\Web Work\doc2vec 最终提交',
'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']
服务器时间:2021 年 5 月 24 日星期一 12:44:47 +0000
D:\Web Work\doc2vec final submit\App\views.py,第171行,在搜索结果中
模型 = Doc2Vec.load("d2vmodel.model")

这是我正在加载 Doc2Vec 模型的 django 函数。

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 模型。
我使用的是 Python 3、PyCharm IDE 并使用 pip 命令安装了库

感谢与问候

1 个答案:

答案 0 :(得分:0)

将模型从App移动到项目的根目录,我认为是'doc2vec final submit'

或者在 'doc2vec final submit' 中创建一个名为 'models' 的文件夹

改变这个

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

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