Skip to content

Latest commit

 

History

History

boston_housing_spark_model

README.md

Boston Housing SparkML Model

This example model uses the Boston Housing Datset to build a sparkML linear regressor model. Then it walks through the steps to on-board this model to Arthur including the steps required to enable the explainability pipeline specifically for sparkML models. All of the required steps can be viewed in the notebook Quickstart.ipynb or in python files which break out the process into three steps, create_model.py, upload_model.py, and send_batch_inference.py.

Setup

Create a virtual environment using your favorite environment management tool and install the requirements. As an example,

python3 -m venv env
source env/bin/activate

pip3 install arthurai
pip3 install -r requirements.txt

Note that the requirements.txt file in this directory assumes python versions 3.6-3.8, as these are currently the only supported versions for the arthur SDK.

Enable Explainability For a SparkML Model:

The steps for uploading the model and sending batch inferences are the same regardless of what type of machine learning model is being used. There are however a few unique steps that must be taken to on-board a SparkML model with explainability. The first step is to create an entrypoint that loads in the SparkML model and implements the predict() function. Unique for a Spark model this file must also initialize a pyspark session in local mode:

from pyspark.sql import SparkSession
spark = SparkSession.builder.master('local').appName('app').getOrCreate()

Optionally the spark session can be configured to use a specified amount of cpu or memory. Whatever configuration is specified in the entrypoint file must also be used to initialize the model server using the sdk's function ArthurModel.enable_explainability(...). An example can be seen below:

model.enable_explainability(df=train_df, project_directory='.',
                            user_predict_function_import_path='entrypoint',
                            requirements_file='requirements.txt',
                            model_server_num_cpu='2',
                            model_server_memory='500Mi')

Note, the minimum amount of cpus which must be allocated for spark models is 2. This must explicitly be defined in the call to enable_explainability(...). The full entrypoint.py example file is provided in this repo. Once these steps are taken and ArthurModel.save() is called, the model will be uploaded and intialized with explainability.

Trouble Shooting:

  1. If you are having issues import pyspark try tunning the jupyter kernel against your virtual environment. To do this run the following command: python -m ipykernel install --user --name=<virtual-env-name> Now when you load a jupyter notebook you should be able to explicitly select what environment to run your kernel in.

  2. If you are running into an error while trying to create a spark session try running the notebook via pyspark with the following commands:

    	export PYSPARK_DRIVER_PYTHON=jupyter
	export PYSPARK_DRIVER_PYTHON_OPTS='notebook'
	pyspark