Я новичок в python, и я надеюсь, что вы можете дать некоторые рекомендации. Мне нравится писать функцию python, чтобы преобразовать ввод данных в формат данных для model.plk.
Пользователь будет вводить 7 переменных, благодаря модели классификатора данные преобразуются через pd. get_dummies для модели. Как я могу написать это в функции?
Например: predict(1,2,1) <- вызвать функцию </p>
, если первый аргумент равен 1 и имеет 7 категорий, первые 7 вводят данные будет (1,0,0,0,0,0,0) Если второй аргумент равен 2 и имеет 5 категорий, входные данные включают в себя первые 7: (1,0,0,0,0,0,0,1,0,0,0) Если третий аргумент равен 1 и имеет 3 категории, входные данные включают первый и второй значения (1,0,0,0,0,0,0,0,1,0,0,0,1,0,0)
Как мне написать это в python функции для использования API? Заранее спасибо.
Текущий
import pandas as pd
import pickle
# load model
model = pickle.load(open('model.pkl','rb'))
def predict(A,B,C):
result = model.predict(x)
# send back to browser
output = {'results': int(result[0])}
# return data
return jsonify(results=output)