databricks-industry-solutions / fraud-orchestration Goto Github PK

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

token = dbutils.notebook.entry_point.getDbutils().notebook().getContext().apiToken().getOrElse(None)
os.environ["DATABRICKS_TOKEN"] = token

def create_tf_serving_json(data):
  return {'inputs': {name: data[name].tolist() for name in data.keys()} if isinstance(data, dict) else data.tolist()}

def score_model(dataset):
  url = 'https://e2-demo-field-eng.cloud.databricks.com/model-endpoint/fraud_xgb_model/137/invocations'
  headers = {'Authorization': f'Bearer {os.environ.get("DATABRICKS_TOKEN")}', 'Content-Type': 'application/json'}
  ds_dict = {'dataframe_split': dataset.to_dict(orient='split')} if isinstance(dataset, pd.DataFrame) else create_tf_serving_json(dataset)
  data_json = json.dumps(ds_dict, allow_nan=True)
  response = requests.request(method='POST', headers=headers, url=url, data=data_json)
  if response.status_code != 200:
    raise Exception(f'Request failed with status {response.status_code}, {response.text}')
  return response.json()

{"dataframe_split":{"index": [0],
 "columns": ["ACCT_PROD_CD",
  "ACCT_AVL_CASH_BEFORE_AMT",
  "ACCT_AVL_MONEY_BEFORE_AMT",
  "ACCT_CL_AMT",
  "ACCT_CURR_BAL",
  "APPRD_AUTHZN_CNT",
  "APPRD_CASH_AUTHZN_CNT",
  "AUTHZN_AMT",
  "AUTHZN_OUTSTD_AMT",
  "AVG_DLY_AUTHZN_AMT",
  "AUTHZN_OUTSTD_CASH_AMT",
  "CDHLDR_PRES_CD",
  "HOTEL_STAY_CAR_RENTL_DUR",
  "LAST_ADR_CHNG_DUR",
  "HOME_PHN_NUM_CHNG_DUR",
  "PLSTC_ISU_DUR",
  "POS_COND_CD",
  "POS_ENTRY_MTHD_CD",
  "DISTANCE_FROM_HOME",
  "FRD_IND"],
 "data": [[6.815292252529644,
   100.0,
   4.897123473461607,
   10000.0,
   8.474660964404574,
   5.2516766208803585,
   2.131421679416424,
   30.0,
   4.169722613481124,
   25.0,
   40.0,
   0.0,
   1.099779462798646,
   301.0,
   5.632847254480254,
   9.010249880517254,
   9.538468313595006,
   1.4806417695062948,
   1000.0,
   2.404762786010208]]}}

An error occured while loading the model. Model Registry features are not supported by the store with URI: 'file:///mlruns'. Stores with the following URI schemes are supported: ['databricks', 'http', 'https', 'postgresql', 'mysql', 'sqlite', 'mssql']..