Wie wenden Sie ein anderes Modell auf verschiedenen Zeilen eines Pandas -Datenframe an?
Posted: 20 Feb 2025, 12:09
Ich habe einen PANDAS -Datenfreame, der so aussieht: < /p>
Ich habe 2 Modelle verschiedene Modelle, die in PKL -Dateien gespeichert sind, die sie so lade:
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
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']})
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)
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)