無法讀取遠程Ubuntu服務器上Flask中保存的機器學習模型

Question

我將機器學習模型保存為“ finalized_model.sav”，正在flask應用程序上使用。 當我在本地桌面上運行以下代碼行時，它運行得很好，但是在遠程Ubuntu服務器上運行相同的代碼卻報錯：

@app.route('/studentData/', methods=['POST'])
def parsing_data():
    try:
        jsonData = request.get_json(force=True)
        data = json.loads(jsonData)  
        requiredData = data['statement']

        filename = 'finalized_model.sav'
        loaded_model = pickle.load(open(filename, 'rb'))
        count_vect = pickle.load(open('count_vect', 'rb'))
        result = loaded_model.predict(count_vect.transform([requiredData]))
        out = {"studentId": str(result)}
        return json.dumps(out)
    except:
        return "invalid data"

我想知道問題可能出在哪里，似乎問題出在加載“ finalized_model.sav”，盡管我將所有文件都保存在燒瓶應用程序（ init .py）所在的同一目錄中

Answer 1

首先，改變

filename = 'finalized_model.sav' 

至

filename = './finalized_model.sav'

如果仍然無法使用，請使用完整路徑名

import os
filename = os.path.abspath(__file__) + os.sep + 'finalized_model.sav'