Python Machine Learning in Power BI

We are going to do some machine learning in Python to transform our dataset into algorithm digestible data for churn analysis. using sci-kit learn It’s a ton easier than it sounds. We will be utilizing the Python scripting option withing in the query editor in Power BI.

There typical stages of machine learning. These steps can be done all within the query editor in Power BI

  1. Preprocessing
  2. Model Training
  3. Predictions

Churn Predictions  with Sci-Kit Learn’s Logistic Regression

The first thing you should do is make a duplicate of your existing dataset. We will be using logistic regression to classify users you have left the company.  This churn is the value that we are trying to predict. We will use all the existing columns as features for our machine learning model to evaluate. Churn Power BI PBIX notebook

Watch the Video Instructions

Features in our data that attribute to someone leaving the company

  • satisfaction level
  • evaluations
  • promotions
  • number of projects,
  • time at the company
  • department
  • accident history
  • promotions

Our Goal

Create a machine learning model that evaluates the features above and provides a new column of  ‘predictions’ and ‘probability of leaving’ to our existing tables.

Our Data

Our data is a mix of categories and numbers which means that we will to encode this data to be presented only as numbers for our machine learning model to understand. We can use the scki-kit preprocessing library to change our data. The data has a ‘left’ column, this is what we will predict.

evaluate the data for machine learning prior to uploading to Power BI


Use the Python Script

These packages will allow us to change the data into numeric and scaled formats. So that we are able to get a more accurate prediction. Additionally, we will split the dataset into training and test sets.

  • Open Power BI
  • Load the dataset, in this example, we will import a csv called  HR_comma_sep
  • Click the Transform heading and then choose Python Script.

user the python script in the edit queries section

Load the Python script window. This is where you will be entering the code. However, I highly encourage to use a IDE when writing your code to ensure the code works, Then copy and paste it into Power BI script editor.

the python script allows to do machine learning in the power bi notebook
The dataset is variable holds the table that you are running the script on. In the Python script using a hashtag is a form of notation to tell the reader what’s going on with your code.

Lets load in the Python dependencies:

import pandas as pd
import numpy as np
from sklearn.preprocessing import LabelEncoder, StandardScaler
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression

Chose your target predictor and features for machine learning:

y = dataset['left']

features = ['satisfaction_level', 'last_evaluation', 'number_project',
'average_montly_hours', 'time_spend_company', 'Work_accident',
'promotion_last_5years', 'Departments ', 'salary']


In machine learning, features are what describes your data aka dependent variable and target is what we are trying to predict.  In this example, I have saved each feature I want to use in my model in a variable called features.

Encode your categories into numbers:

le = LabelEncoder()
df['salary'] = le.fit_transform(df['salary'])
df['Departments'] = le.fit_transform(df['Departments'])

Split the Data  and Train the Model:

X_train,X_test,y_train,y_test = train_test_split(X,y)

Use the Logistic Regression Model Prediction

log = LogisticRegression(),y_train)

Add the columns back to the dataframe(table)

dataset['predictions'] = y_pred
dataset['probability of leaving'] = y_prob[:,1]

The Full Code(Copy and Paste)–Bringing it All Together!

#Load in the dependencies
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.preprocessing import LabelEncoder, StandardScaler

#lets change categories to numbers
le = LabelEncoder()
dataset['Departments'] = le.fit_transform(dataset['Departments'])
dataset['salary'] = le.fit_transform(dataset['salary'])
#preprocess your data
features = ['satisfaction_level', 'last_evaluation', 'number_project',
'average_montly_hours', 'time_spend_company', 'Work_accident',
'promotion_last_5years', 'Departments', 'salary']
#lets scale the data
s = StandardScaler()
X = s.fit_transform(X)

#split and train the dataset
X_train,X_test,y_train,y_test = train_test_split(X,y)

#Let the model predict results
log = LogisticRegression(),y_train)
y_pred = log.predict(X)
y_prob = log.predict_proba(X)

# Lets add the columns back to the dataframe
dataset['predictions'] = y_pred
dataset['probability of leaving'] = y_prob[:,1]


Code Added to your Script Box

Once everything is complete you will be left with a result new table which will show the modified results made to your original table. This will be saved as a new query called dataset.

Click on the table to get the result changes which would be our two additional tables added  ‘Prediction’ and ‘Probability of Leaving’. We can use these tables to judge the accuracy of our model.


User Power BI to visualize the model performance by comparing actual churn to your model’s performance. You improve the model’s performance through tunning or using other algorithms such as gradient boosting or cat boosting algorithms. Also, you can change the thresholds.

use power bi to visualize your models performance


Check out how to Use Python Visuals in Power BI

About the Author

Leave a Reply

9 Comments on "Python Machine Learning in Power BI"

Notify of
Sort by:   newest | oldest | most voted

Powerbi error: ValueError: could not convert string to float: ‘s’


Before ‘X=dataset[features]’

You need to apply this line to all of your string type feature (I added the “.astype(str)” as well):
df[‘feature1’] = le.fit_transform(df[‘feature1’].astype(str))
df[‘feature2’] = le.fit_transform(df[‘feature2’].astype(str))

Also get rid of null values (Do a replace of null with “NA” or something

hi, got below when copied the full code into power bi ADO.NET: Python script error. Traceback (most recent call last): File “PythonScriptWrapper.PY”, line 14, in from sklearn.model_selection import train_test_split File “D:\PYTHON\lib\site-packages\sklearn\”, line 134, in from .base import clone File “D:\PYTHON\lib\site-packages\sklearn\”, line 13, in from .utils.fixes import signature File “D:\PYTHON\lib\site-packages\sklearn\utils\”, line 10, in from .validation import (as_float_array, File “D:\PYTHON\lib\site-packages\sklearn\utils\”, line 18, in from ..utils.fixes import signature File “D:\PYTHON\lib\site-packages\sklearn\utils\”, line 144, in from scipy.sparse.linalg import lsqr as sparse_lsqr # noqa File “D:\PYTHON\lib\site-packages\scipy\sparse\linalg\”, line 114, in from .isolve import * File “D:\PYTHON\lib\site-packages\scipy\sparse\linalg\isolve\”, line 6, in from .iterative import * File “D:\PYTHON\lib\site-packages\scipy\sparse\linalg\isolve\”, line 10,… Read more »