INNER CODE UNIT · Python
get_model
featurestoreorg/serverless-ml-course · src/04-module/cc-fraud-streamlit-ui.py:50
def get_model(project = project):
mr = project.get_model_registry()
model = mr.get_model("cc_fraud", version = 1)
model_dir = model.download()
return joblib.load(model_dir + "/cc_fraud_model.pkl")
def explore_data(batch_data):
st.write(36 * "-")
print_fancy_header('\n👁 Data Exploration...')
labels = ["Suspected of Fraud", "Not Suspected of Fraud"]
unique, counts = np.unique(batch_data.fraud.values, return_counts=True)
values = counts.tolist()
def plot_pie(values, labels):
fig = px.pie(values=values, names=labels, title='Distribution of predicted fraud transactions')
return fig
fig1 = plot_pie(values, labels)