Wie wenden Sie ein anderes Modell auf verschiedenen Zeilen eines Pandas -Datenframe an?Python

Python-Programme
Guest
 Wie wenden Sie ein anderes Modell auf verschiedenen Zeilen eines Pandas -Datenframe an?

Post by Guest »

Ich habe einen PANDAS -Datenfreame, der so aussieht: < /p>

Code: Select all

import pandas as pd
df = pd.DataFrame({'id': [1,2], 'var1': [5,6], 'var2': [20,60], 'var3': [8, -2], 'model_version': ['model_a', 'model_b']})
Ich habe 2 Modelle verschiedene Modelle, die in PKL -Dateien gespeichert sind, die sie so lade:

Code: Select all

import pickle
with open('model_a.pkl', 'rb') as file:
model_a = pickle.load(file)
with open('model_b.pkl', 'rb') as file:
model_b = pickle.load(file)
Ich möchte model_a für id == 1 und model_b für id == 2 anwenden und vorherzusagen, wie angegeben In der Spalte model_version im df .
Wie kann ich das in "Ein-Go"? Was ich damit meine "Ein-Go" ist das: Ich bin nicht Auf der Suche nach einer Lösung, die so aussieht:

Code: Select all

df_a = df.query('model_version=="model_a"')
predictions_a = model_a.predict(df_a)

df_b = df.query('model_version=="model_b"')
predictions_b = model_b.predict(df_b)

Quick Reply

Change Text Case: 
   
  • Similar Topics
    Replies
    Views
    Last post