Edit model card

Breast Cancer Prediction with Decision Trees 🌲 🌳

This is a decision tree model trained on breast cancer prediction dataset.

Plot of the Tree

Below is the visualization of tree. 🌲 tree

Feature Importances

You can find feature importances below. πŸ‘‡πŸΌ imp

Training Hyperparameters

Find hyperparameters of the model below πŸ“‘

0
ccp_alpha 0.0
class_weight
criterion gini
max_depth
max_features
max_leaf_nodes
min_impurity_decrease 0.0
min_impurity_split
min_samples_leaf 1
min_samples_split 2
min_weight_fraction_leaf 0.0
random_state 0
splitter best

Metrics

Find metrics of the model below πŸ“ˆ

0 1 accuracy macro avg weighted avg
precision 0.907407 0.955056 0.937063 0.931232 0.937396
recall 0.924528 0.944444 0.937063 0.934486 0.937063
f1-score 0.915888 0.949721 0.937063 0.932804 0.937181
support 53 90 0.937063 143 143

Try it yourself!

You can use this model with

from huggingface_hub import hf_hub_download
import joblib

model = joblib.load(
    hf_hub_download("merve/cancer-prediction", "sklearn_model.joblib")
)
Downloads last month
0
Hosted inference API
This model can be loaded on the Inference API on-demand.