Получение KeyError в Klepto при использовании с локальной машины - PullRequest
Новая
       17

Получение KeyError в Klepto при использовании с локальной машины

0 голосов
/ 23 октября 2019

Я хочу сохранить мою модель ML на моей локальной машине. Я следовал https://stackoverflow.com/a/29291153/11543494 этому ответу, чтобы сохранить модель ML на локальной машине, но когда я загружаю постоянную модель ML с локальной машины, я получаю ключевую ошибку.

Я создал модель ML, которая будет предсказыватьКатегория URL. Теперь я хочу интегрировать модель ML с моим веб-приложением, и поэтому я создал API с помощью Flask.

Я протестировал свою модель ML в блокноте jupyter, где у меня сейчас есть код, связанный с моделью ML, я просто хочучтобы сбросить мою модель ML и использовать ее в моем API. В блокноте jupyter я получаю правильный вывод, но когда я загружаю постоянный файл в мой код API, я получаю KeyError. Я попытался использовать pickle, joblib, но там я получаю MemoryError, я тоже пытался решить эту проблему, но мне не удалось решить эту проблему, поэтому я пытаюсь Klepto.

Код Klepto 

from klepto.archives import dir_archive

model = dir_archive('E:/Mayur/Sem 5/Python project/model_klepto',{'result':gs_clf},serialized=True, cached=False)
#gs_clf = gs_clf.fit(x_train, y_train) #RandomizedSearchCV
model.dump()

Код API 

import numpy as np
from flask import Flask, request, jsonify, render_template
from klepto.archives import dir_archive

app = Flask(__name__)

demo = dir_archive(
    'E:/Mayur/Sem 5/Python project/model_klepto', {}, serialized=True, cached=False)
demo.load()


@app.route('/')
def home():
    return render_template('index.html')


@app.route('/predict', methods=['POST'])
def predict():
    input = request.form.values()
    final_feature = [np.array(input)]
    prediction = demo['result'].predict([str(final_feature)])

    return render_template('index.html', prediction_text=prediction)


if __name__ == "__main__":
    app.run(debug=True)

Когда я запускаю API, я получаю KeyError: 'result'.

Если я запускаюниже код в блокноте Jupyter, я получаю правильный вывод 

demo = dir_archive(
    'E:/Mayur/Sem 5/Python project/model_klepto', {}, serialized=True, cached=False)
demo.load()

demo

Выход> 

ir_archive('model_klepto', {'result': RandomizedSearchCV(cv='warn', error_score='raise-deprecating',
                   estimator=Pipeline(memory=None,
                                      steps=[('vect',
                                              CountVectorizer(analyzer='word',
                                                              binary=False,
                                                              decode_error='strict',
                                                              dtype=<class 'numpy.int64'>,
                                                              encoding='utf-8',
                                                              input='content',
                                                              lowercase=True,
                                                              max_df=1.0,
                                                              max_features=None,
                                                              min_df=1,
                                                              ngram_range=(1,
                                                                           1),
                                                              preprocessor=None,
                                                              stop_words=None,
                                                              strip_accen...
                                                               sublinear_tf=False,
                                                               use_idf=True)),
                                             ('clf',
                                              MultinomialNB(alpha=1.0,
                                                            class_prior=None,
                                                            fit_prior=True))],
                                      verbose=False),
                   iid='warn', n_iter=5, n_jobs=None,
                   param_distributions={'clf__alpha': (0.01, 0.001),
                                        'tfidf__use_idf': (True, False),
                                        'vect__ngram_range': [(1, 1), (1, 2)]},
                   pre_dispatch='2*n_jobs', random_state=None, refit=True,
                   return_train_score=False, scoring=None, verbose=0)}, cached=False)

demo['result'].predict(['http://www.windows.com'])

Выход> массив (['Computers'], dtype =

Вот скриншот трассировки стека Трассировка стека

...