bhavesh's picture
Upload ./ with huggingface_hub
05e2b8a
metadata
license: apache-2.0
library_name: sklearn
tags:
  - tabular-classification
  - baseline-trainer

Baseline Model trained on arinfo_sample_dataset_finaltffwjv58 to apply classification on model

Metrics of the best model:

accuracy 0.930688

recall_macro 0.655991

precision_macro 0.640972

f1_macro 0.638021

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

See model plot below:

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

rto False False False ... False True False ownerNum False False False ... False False False cc False False False ... False False False insurance False False False ... False False False weight True False False ... False False False financer False False False ... False True False fu... class False False False ... False False False state False False False ... False False False year False False False ... False False False categoryId False False False ... False False False onroadPrice True False False ... False False False price_FAIR True False False ... False False False[13 rows x 7 columns])),('decisiontreeclassifier',DecisionTreeClassifier(class_weight='balanced',max_depth=2249))])

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.

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