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)