Baseline Model trained on UCI_Credit_Cardyi6q1ptm to apply classification on PAY_0
Metrics of the best model:
accuracy 0.715467
recall_macro 0.777916
precision_macro 0.578960
f1_macro 0.596625
Name: DecisionTreeClassifier(class_weight='balanced', min_impurity_decrease=0.01), dtype: float64
See model plot below:
Pipeline(steps=[('easypreprocessor',EasyPreprocessor(types= continuous dirty_float ... free_string useless
LIMIT_BAL False False ... False False
SEX False False ... False False
EDUCATION False False ... False False
MARRIAGE False False ... False False
AGE False False ... False False
PAY_2 False False ... False False
PAY_3 False False ... False False
PAY_4 False False ... False False
PAY_5 False False ......
PAY_AMT1 True False ... False False
PAY_AMT2 True False ... False False
PAY_AMT3 True False ... False False
PAY_AMT4 True False ... False False
PAY_AMT5 True False ... False False
PAY_AMT6 True False ... False False
default.payment.next.month False False ... False False[23 rows x 7 columns])),('decisiontreeclassifier',DecisionTreeClassifier(class_weight='balanced',min_impurity_decrease=0.01))])
In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook. On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.
Pipeline(steps=[('easypreprocessor',EasyPreprocessor(types= continuous dirty_float ... free_string useless
LIMIT_BAL False False ... False False
SEX False False ... False False
EDUCATION False False ... False False
MARRIAGE False False ... False False
AGE False False ... False False
PAY_2 False False ... False False
PAY_3 False False ... False False
PAY_4 False False ... False False
PAY_5 False False ......
PAY_AMT1 True False ... False False
PAY_AMT2 True False ... False False
PAY_AMT3 True False ... False False
PAY_AMT4 True False ... False False
PAY_AMT5 True False ... False False
PAY_AMT6 True False ... False False
default.payment.next.month False False ... False False[23 rows x 7 columns])),('decisiontreeclassifier',DecisionTreeClassifier(class_weight='balanced',min_impurity_decrease=0.01))])EasyPreprocessor(types= continuous dirty_float ... free_string useless LIMIT_BAL False False ... False False SEX False False ... False False EDUCATION False False ... False False MARRIAGE False False ... False False AGE False False ... False False PAY_2 False False ... False False PAY_3 False False ... False False PAY_4 False False ... False False PAY_5 False False ... False False PAY_6 False False ... False Fal... BILL_AMT3 True False ... False False BILL_AMT4 True False ... False False BILL_AMT5 True False ... False False BILL_AMT6 True False ... False False PAY_AMT1 True False ... False False PAY_AMT2 True False ... False False PAY_AMT3 True False ... False False PAY_AMT4 True False ... False False PAY_AMT5 True False ... False False PAY_AMT6 True False ... False False default.payment.next.month False False ... False False[23 rows x 7 columns])
DecisionTreeClassifier(class_weight='balanced', min_impurity_decrease=0.01)
Disclaimer: This model is trained with dabl library as a baseline, for better results, use AutoTrain.
Logs of training including the models tried in the process can be found in logs.txt
- Downloads last month
- 0