Как применить .pkl (модель ML) ко всему фрейму данных - PullRequest
Новая
       54

Как применить .pkl (модель ML) ко всему фрейму данных

0 голосов
/ 27 мая 2020

У меня есть модель в форме рассола (model1.pkl), и мне нужно спрогнозировать новый файл excel, который был мне предоставлен. В настоящее время я использую от pandas до l oop для каждой строки и получаю ответ 

df=pd.read_excel("data.xlsx") 
model1 = pickle.load(open('model1.pkl', 'rb'))
appts=(list(df['apptid'].unique()))

for i in appts:
    fdf = df[df['apptid'] == i]
    fdf.fillna(0, inplace=True)
    fdf.reset_index(inplace=True)
    hdf = fdf
    fdf = fdf.iloc[:1]
    ans = model1.predict(
        [[fdf['a'][0], fdf['b'][0], fdf['c'][0],
          fdf['d'][0], fdf['e'][0], fdf['f'][0], fdf['g'][0]]])
    fdf['ans'] = int(round(ans[0]))
    print("anssss", ans[0])

Есть ли другой способ, при котором мне не нужно l oop и можно напрямую предсказать для всех строк, доступных в Excel ??

...