Prediction script example

predict.py

import json
import os
import numpy as np
import pandas as pd
import joblib

# Your imports go here
  
# Update to your model's filename
model_filename = "model.pkl"
model_dir = "./"

print("Model dir:", model_dir)
print("Model filename:", model_filename)
    
model_path = os.path.join(model_dir, model_filename)

# Replace this line with your model loading code
model = joblib.load(model_path)

# Adapt this code with some sample data and also adapt prediction step
data = {...} # define some sample data
df = pd.DataFrame.from_dict(data)
proba = model.predict_proba(df)

result = {"predict_proba": proba.tolist()}
print(result)