Starter Example
This is a basic example of explainX Open-Source usage in explaining a Random Forest model.
After successfully installing explainX, open up your Python IDE of Jupyter Notebook and simply follow the code below to use it:
Open up your jupyter notebook and simply follow the code.
Goal
The goal is to be able to explain and debug the machine learning model we are building. The main goal will be to provide business-level explanations.
Prerequisite
Make sure you have Sklearn & explainX installed. In case you don't, follow the following command to install the libraries.
- 1.Import required module.
1
from explainx import *
2
from sklearn.ensemble import RandomForestClassifier
3
from sklearn.model_selection import train_test_split
Copied!
- 1.Load and split your dataset into x_data and y_data
1
#Load Dataset: X_Data, Y_Data
2
#X_Data = Pandas DataFrame
3
#Y_Data = Numpy Array or List
4
5
X_data,Y_data = explainx.dataset_heloc()
Copied!
- 1.Split dataset into training & testing.
1
X_train, X_test, Y_train, Y_test = train_test_split(X_data,Y_data, test_size=0.3, random_state=0)
Copied!
- 1.Train your model.
1
# Train a RandomForest Model
2
model = RandomForestClassifier()
3
model.fit(X_train, Y_train)
Copied!
After you're done training the model, you can either access the complete explainability dashboard or access individual techniques.
Complete Explainability Dashboard
To access the entire dashboard with all the explainability techniques under one roof, follow the code down below. It is great for sharing your work with your peers and managers in an interactive and easy to understand way.
5.1. Pass your model and dataset into the explainX function:
1
explainx.ai(X_test, Y_test, model, model_name="randomforest")
Copied!
5.2. Click on the dashboard link to start exploring model behavior:
1
App running on https://0.0.0.0:8080
Copied!
Explainability Modules
In this latest release, we have also given the option to use explainability techniques individually. This will allow the user to choose technique that fits their personal AI use case.
6.1. Pass your model, X_Data and Y_Data into the explainx_modules function.
1
explainx_modules.ai(X_test, Y_test, model)
Copied!
As an upgrade, we have eliminated the need to pass in the model name as explainX is smart enough to identify the model type and problem type i.e. classification or regression, by itself.
You can access multiple modules:
Module 1: Dataframe with Predictions
1
explainx_modules.dataframe_graphing()
Copied!
Module 2: Model Metrics
1
explainx_modules.metrics()
Copied!
Module 3: Global Level SHAP Values
1
explainx_modules.shap_df()
Copied!
Module 4: What-If Scenario Analysis (Local Level Explanations)
1
explainx_modules.what_if_analysis()
Copied!
Module 5: Partial Dependence Plot & Summary Plot
1
explainx_modules.feature_interactions()
Copied!
Module 6: Model Performance Comparison (Cohort Analysis)
1
explainx_modules.cohort_analysis()
Copied!
To access the modules within your jupyter notebook as IFrames, just pass the mode='inline' argument in each of the function.
Last modified 11mo ago
Export as PDF
Copy link