Machine Learning for Marketing in Python
Karolis Urbonas
Head of Analytics & Science, Amazon
Main churn typology is based on two business model types:
Typically:
Churn
/No Churn
or Yes
/No
- best practice to transform as 1 and 0set(telcom['Churn'])
{0, 1}
telcom.groupby(['Churn']).size() / telcom.shape[0] * 100
Churn
0 73.421502
1 26.578498
dtype: float64
from sklearn.model_selection import train_test_split
train, test = train_test_split(telcom, test_size = .25)
Separate column names by data types
target = ['Churn']
custid = ['customerID']
cols = [col for col in telcom.columns if col not in custid + target]
Build training and testing datasets
train_X = train[cols]
train_Y = train[target]
test_X = test[cols]
test_Y = test[target]
Machine Learning for Marketing in Python