You can deploy this model and start earning money today!

Testys/
diabetespmxrsn1x-diabetes-classification

Name: diabetespmxrsn1x-diabetes-classification
Availability: InStock

sklearn

tabular-classification

baseline-trainer

apache-2.0

Views

No views yet

Baseline Model trained on diabetespmxrsn1x to apply classification on diabetes

Metrics of the best model:

accuracy 0.871795

average_precision 0.518856

roc_auc 0.883333

recall_macro 0.883333

f1_macro 0.801996

Name: DecisionTreeClassifier(class_weight='balanced', max_depth=1), dtype: float64

See model plot below:

Pipeline(steps=[('easypreprocessor',EasyPreprocessor(types= continuous dirty_float ... free_string useless

cholesterol True False ... False False glucose True False ... False False hdl_chol True False ... False False chol_hdl_ratio False False ... True False age True False ... False False gender False False ... False False height False False ... False False weight True False ... False False bmi False False ... True False systolic_bp True False ... False False diastolic_bp True False ... False False waist False False ... False False hip False False ... False False waist_hip_ratio False False ... False False[14 rows x 7 columns])),('decisiontreeclassifier',DecisionTreeClassifier(class_weight='balanced', max_depth=1))])

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

Pipeline(steps=[('easypreprocessor',EasyPreprocessor(types= continuous dirty_float ... free_string useless cholesterol True False ... False False glucose True False ... False False hdl_chol True False ... False False chol_hdl_ratio False False ... True False age True False ... False False gender False False ... False False height False False ... False False weight True False ... False False bmi False False ... True False systolic_bp True False ... False False diastolic_bp True False ... False False waist False False ... False False hip False False ... False False waist_hip_ratio False False ... False False[14 rows x 7 columns])),('decisiontreeclassifier',DecisionTreeClassifier(class_weight='balanced', max_depth=1))])

EasyPreprocessor

EasyPreprocessor(types= continuous dirty_float ... free_string useless cholesterol True False ... False False glucose True False ... False False hdl_chol True False ... False False chol_hdl_ratio False False ... True False age True False ... False False gender False False ... False False height False False ... False False weight True False ... False False bmi False False ... True False systolic_bp True False ... False False diastolic_bp True False ... False False waist False False ... False False hip False False ... False False waist_hip_ratio False False ... False False[14 rows x 7 columns])

DecisionTreeClassifier

DecisionTreeClassifier(class_weight='balanced', max_depth=1)

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

Testys/diabetespmxrsn1x-diabetes-classification

Baseline Model trained on diabetespmxrsn1x to apply classification on diabetes

Testys/
diabetespmxrsn1x-diabetes-classification