metadata

library_name: sklearn
license: mit
tags:
  - sklearn
  - skops
  - tabular-classification
model_format: pickle
model_file: example_model_v1.pkl
widget:
  - structuredData:
      Academic Pressure:
        - -0.04606674611568451
        - -0.04606674611568451
        - -0.04606674611568451
      Academic Pressure_na:
        - 2
        - 2
        - 2
      Age:
        - 1.2607895135879517
        - -0.19406381249427795
        - 1.584090232849121
      CGPA:
        - 0.03309895098209381
        - 0.03309895098209381
        - 0.03309895098209381
      CGPA_na:
        - 2
        - 2
        - 2
      City:
        - 98
        - 28
        - 27
      Degree:
        - 77
        - 90
        - 77
      Dietary Habits:
        - 16
        - 21
        - 21
      Family History of Mental Illness:
        - 1
        - 2
        - 2
      Financial Stress:
        - 1.4229286909103394
        - 1.4229286909103394
        - 0.008482186123728752
      Financial Stress_na:
        - 1
        - 1
        - 1
      Gender:
        - 2
        - 2
        - 2
      Have you ever had suicidal thoughts ?:
        - 1
        - 1
        - 2
      Job Satisfaction:
        - 0.8020281791687012
        - 1.5897727012634277
        - -1.5612051486968994
      Job Satisfaction_na:
        - 1
        - 1
        - 1
      Name:
        - 79
        - 269
        - 90
      Profession:
        - 46
        - 15
        - 56
      Sleep Duration:
        - 30
        - 20
        - 30
      Study Satisfaction:
        - 0.019511962309479713
        - 0.019511962309479713
        - 0.019511962309479713
      Study Satisfaction_na:
        - 2
        - 2
        - 2
      Work Pressure:
        - 1.588835597038269
        - -0.7938991189002991
        - -0.7938991189002991
      Work Pressure_na:
        - 1
        - 1
        - 1
      Work/Study Hours:
        - 1.2296171188354492
        - 1.4891586303710938
        - -1.1062564849853516
      Working Professional or Student:
        - 2
        - 2
        - 2

Model description

[More Information Needed]

Intended uses & limitations

[More Information Needed]

Training Procedure

[More Information Needed]

Hyperparameters

Click to expand

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

Model Plot

DecisionTreeClassifier()

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.

Evaluation Results

Metric	Value
accuracy	0.899751

How to Get Started with the Model

[More Information Needed]

Model Card Authors

This model card is written by following authors:

[More Information Needed]

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]

citation_bibtex

bibtex @inproceedings{...,year={2024}}

get_started_code

import pickle with open(dtc_pkl_filename, 'rb') as file: clf = pickle.load(file)

model_card_authors

Rubanza

limitations

This model is not ready to be used in production.

model_description

This is a random forest classifier model trained a mental health dataset.

eval_method

The model is evaluated using test split, on accuracy