metadata

library_name: sklearn
tags:
  - sklearn
  - skops
  - tabular-classification
widget:
  structuredData:
    attribute_0:
      - material_7
      - material_7
      - material_7
    attribute_1:
      - material_8
      - material_8
      - material_5
    attribute_2:
      - 9
      - 9
      - 6
    attribute_3:
      - 5
      - 5
      - 6
    loading:
      - 150.15
      - 106.3
      - 117.52
    measurement_0:
      - 6
      - 11
      - 4
    measurement_1:
      - 7
      - 4
      - 9
    measurement_10:
      - 15.888
      - 15.56
      - 18.49
    measurement_11:
      - 21.623
      - 17.233
      - 20.193
    measurement_12:
      - 12.83
      - 12.926
      - 14.127
    measurement_13:
      - 14.738
      - 14.367
      - 15.185
    measurement_14:
      - 18.506
      - 16.302
      - 16.657
    measurement_15:
      - 14.16
      - 15.018
      - 13.326
    measurement_16:
      - 15.266
      - 18.297
      - 17.467
    measurement_17:
      - 674.165
      - 604.836
      - 648.023
    measurement_2:
      - 11
      - 4
      - 9
    measurement_3:
      - 19.637
      - 18.217
      - 19.325
    measurement_4:
      - 12.55
      - 10.627
      - 10.092
    measurement_5:
      - 17.119
      - 17.74
      - 17.218
    measurement_6:
      - NaN
      - 17.295
      - 17.962
    measurement_7:
      - 10.958
      - 11.732
      - 9.274
    measurement_8:
      - 17.93
      - 17.591
      - 18.653
    measurement_9:
      - NaN
      - 12.689
      - 13.149
    product_code:
      - A
      - A
      - D

Model description

This is a DecisionTreeClassifier model built for Kaggle Tabular Playground Series August 2022, trained on supersoaker production failures dataset.

Intended uses & limitations

This model is not ready to be used in production.

Training Procedure

Hyperparameters

The model is trained with below hyperparameters.

Click to expand

Hyperparameter	Value
memory
steps	[('transformation', ColumnTransformer(transformers=[('loading_missing_value_imputer',

                             SimpleImputer(), ['loading']),
                            ('numerical_missing_value_imputer',
                             SimpleImputer(),
                             ['loading', 'measurement_3', 'measurement_4',
                              'measurement_5', 'measurement_6',
                              'measurement_7', 'measurement_8',
                              'measurement_9', 'measurement_10',
                              'measurement_11', 'measurement_12',
                              'measurement_13', 'measurement_14',
                              'measurement_15', 'measurement_16',
                              'measurement_17']),
                            ('attribute_0_encoder', OneHotEncoder(),
                             ['attribute_0']),
                            ('attribute_1_encoder', OneHotEncoder(),
                             ['attribute_1']),
                            ('product_code_encoder', OneHotEncoder(),
                             ['product_code'])])), ('model', DecisionTreeClassifier(max_depth=4))]                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                |

Model Plot

The model plot is below.

Evaluation Results

You can find the details about evaluation process and the evaluation results.

Metric	Value
accuracy	0.778564
f1 score	0.778564

How to Get Started with the Model

Use the code below to get started with the model.

Click to expand

import pickle 
with open(decision-tree-playground-kaggle/model.pkl, 'rb') as file: 
    clf = pickle.load(file)

Model Card Authors

This model card is written by following authors:

huggingface

Model Card Contact

You can contact the model card authors through following channels: [More Information Needed]

Citation

Below you can find information related to citation.

BibTeX:

[More Information Needed]

Tree Plot

Confusion Matrix