|
--- |
|
license: mit |
|
library_name: sklearn |
|
tags: |
|
- sklearn |
|
- skops |
|
- tabular-classification |
|
model_format: pickle |
|
model_file: catboost_without_hospital_number.pkl |
|
widget: |
|
structuredData: |
|
abdomen: |
|
- distend_small |
|
- distend_small |
|
- distend_large |
|
abdominal_distention: |
|
- none |
|
- none |
|
- moderate |
|
abdomo_appearance: |
|
- serosanguious |
|
- cloudy |
|
- serosanguious |
|
abdomo_protein: |
|
- 4.1 |
|
- 4.3 |
|
- 2.0 |
|
age: |
|
- adult |
|
- adult |
|
- adult |
|
capillary_refill_time: |
|
- less_3_sec |
|
- less_3_sec |
|
- more_3_sec |
|
cp_data: |
|
- 'yes' |
|
- 'yes' |
|
- 'no' |
|
lesion_1: |
|
- 7209 |
|
- 2112 |
|
- 5400 |
|
lesion_2: |
|
- 0 |
|
- 0 |
|
- 0 |
|
lesion_3: |
|
- 0 |
|
- 0 |
|
- 0 |
|
mucous_membrane: |
|
- bright_pink |
|
- bright_pink |
|
- dark_cyanotic |
|
nasogastric_reflux: |
|
- none |
|
- none |
|
- more_1_liter |
|
nasogastric_reflux_ph: |
|
- 7.0 |
|
- 3.5 |
|
- 2.0 |
|
nasogastric_tube: |
|
- slight |
|
- none |
|
- significant |
|
packed_cell_volume: |
|
- 37.0 |
|
- 44.0 |
|
- 65.0 |
|
pain: |
|
- depressed |
|
- mild_pain |
|
- extreme_pain |
|
peripheral_pulse: |
|
- normal |
|
- normal |
|
- reduced |
|
peristalsis: |
|
- hypermotile |
|
- hypomotile |
|
- absent |
|
pulse: |
|
- 84.0 |
|
- 66.0 |
|
- 72.0 |
|
rectal_exam_feces: |
|
- absent |
|
- decreased |
|
- absent |
|
rectal_temp: |
|
- 39.0 |
|
- 38.5 |
|
- 37.3 |
|
respiratory_rate: |
|
- 24.0 |
|
- 21.0 |
|
- 30.0 |
|
surgery: |
|
- 'yes' |
|
- 'yes' |
|
- 'yes' |
|
surgical_lesion: |
|
- 'yes' |
|
- 'yes' |
|
- 'yes' |
|
temp_of_extremities: |
|
- cool |
|
- normal |
|
- cool |
|
total_protein: |
|
- 6.5 |
|
- 7.6 |
|
- 13.0 |
|
--- |
|
|
|
# Model description |
|
|
|
This is a `Catboost` model trained on horse health outcome data from Kaggle. |
|
|
|
## Intended uses & limitations |
|
|
|
This model is not ready to be used in production. |
|
|
|
## Training Procedure |
|
|
|
[More Information Needed] |
|
|
|
### Hyperparameters |
|
|
|
<details> |
|
<summary> Click to expand </summary> |
|
|
|
| Hyperparameter | Value | |
|
|--------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------| |
|
| memory | | |
|
| steps | [('preprocessor', ColumnTransformer(remainder='passthrough',<br /> transformers=[('num',<br /> Pipeline(steps=[('imputer',<br /> SimpleImputer(strategy='median')),<br /> ('scaler', StandardScaler())]),<br /> ['rectal_temp', 'pulse', 'respiratory_rate',<br /> 'nasogastric_reflux_ph', 'packed_cell_volume',<br /> 'total_protein', 'abdomo_protein', 'lesion_1',<br /> 'lesion_2', 'lesion_3']),<br /> ('cat',<br /> Pipeline(steps=[('imputer',<br /> SimpleI...='missing',<br /> strategy='constant')),<br /> ('onehot',<br /> OneHotEncoder(handle_unknown='ignore'))]),<br /> ['surgery', 'age', 'temp_of_extremities',<br /> 'peripheral_pulse', 'mucous_membrane',<br /> 'capillary_refill_time', 'pain',<br /> 'peristalsis', 'abdominal_distention',<br /> 'nasogastric_tube', 'nasogastric_reflux',<br /> 'rectal_exam_feces', 'abdomen',<br /> 'abdomo_appearance', 'surgical_lesion',<br /> 'cp_data'])])), ('classifier', <catboost.core.CatBoostClassifier object at 0x000001C4CE4ABF10>)] | |
|
| verbose | False | |
|
| preprocessor | ColumnTransformer(remainder='passthrough',<br /> transformers=[('num',<br /> Pipeline(steps=[('imputer',<br /> SimpleImputer(strategy='median')),<br /> ('scaler', StandardScaler())]),<br /> ['rectal_temp', 'pulse', 'respiratory_rate',<br /> 'nasogastric_reflux_ph', 'packed_cell_volume',<br /> 'total_protein', 'abdomo_protein', 'lesion_1',<br /> 'lesion_2', 'lesion_3']),<br /> ('cat',<br /> Pipeline(steps=[('imputer',<br /> SimpleI...='missing',<br /> strategy='constant')),<br /> ('onehot',<br /> OneHotEncoder(handle_unknown='ignore'))]),<br /> ['surgery', 'age', 'temp_of_extremities',<br /> 'peripheral_pulse', 'mucous_membrane',<br /> 'capillary_refill_time', 'pain',<br /> 'peristalsis', 'abdominal_distention',<br /> 'nasogastric_tube', 'nasogastric_reflux',<br /> 'rectal_exam_feces', 'abdomen',<br /> 'abdomo_appearance', 'surgical_lesion',<br /> 'cp_data'])]) | |
|
| classifier | <catboost.core.CatBoostClassifier object at 0x000001C4CE4ABF10> | |
|
| preprocessor__n_jobs | | |
|
| preprocessor__remainder | passthrough | |
|
| preprocessor__sparse_threshold | 0.3 | |
|
| preprocessor__transformer_weights | | |
|
| preprocessor__transformers | [('num', Pipeline(steps=[('imputer', SimpleImputer(strategy='median')),<br /> ('scaler', StandardScaler())]), ['rectal_temp', 'pulse', 'respiratory_rate', 'nasogastric_reflux_ph', 'packed_cell_volume', 'total_protein', 'abdomo_protein', 'lesion_1', 'lesion_2', 'lesion_3']), ('cat', Pipeline(steps=[('imputer',<br /> SimpleImputer(fill_value='missing', strategy='constant')),<br /> ('onehot', OneHotEncoder(handle_unknown='ignore'))]), ['surgery', 'age', 'temp_of_extremities', 'peripheral_pulse', 'mucous_membrane', 'capillary_refill_time', 'pain', 'peristalsis', 'abdominal_distention', 'nasogastric_tube', 'nasogastric_reflux', 'rectal_exam_feces', 'abdomen', 'abdomo_appearance', 'surgical_lesion', 'cp_data'])] | |
|
| preprocessor__verbose | False | |
|
| preprocessor__verbose_feature_names_out | True | |
|
| preprocessor__num | Pipeline(steps=[('imputer', SimpleImputer(strategy='median')),<br /> ('scaler', StandardScaler())]) | |
|
| preprocessor__cat | Pipeline(steps=[('imputer',<br /> SimpleImputer(fill_value='missing', strategy='constant')),<br /> ('onehot', OneHotEncoder(handle_unknown='ignore'))]) | |
|
| preprocessor__num__memory | | |
|
| preprocessor__num__steps | [('imputer', SimpleImputer(strategy='median')), ('scaler', StandardScaler())] | |
|
| preprocessor__num__verbose | False | |
|
| preprocessor__num__imputer | SimpleImputer(strategy='median') | |
|
| preprocessor__num__scaler | StandardScaler() | |
|
| preprocessor__num__imputer__add_indicator | False | |
|
| preprocessor__num__imputer__copy | True | |
|
| preprocessor__num__imputer__fill_value | | |
|
| preprocessor__num__imputer__keep_empty_features | False | |
|
| preprocessor__num__imputer__missing_values | nan | |
|
| preprocessor__num__imputer__strategy | median | |
|
| preprocessor__num__scaler__copy | True | |
|
| preprocessor__num__scaler__with_mean | True | |
|
| preprocessor__num__scaler__with_std | True | |
|
| preprocessor__cat__memory | | |
|
| preprocessor__cat__steps | [('imputer', SimpleImputer(fill_value='missing', strategy='constant')), ('onehot', OneHotEncoder(handle_unknown='ignore'))] | |
|
| preprocessor__cat__verbose | False | |
|
| preprocessor__cat__imputer | SimpleImputer(fill_value='missing', strategy='constant') | |
|
| preprocessor__cat__onehot | OneHotEncoder(handle_unknown='ignore') | |
|
| preprocessor__cat__imputer__add_indicator | False | |
|
| preprocessor__cat__imputer__copy | True | |
|
| preprocessor__cat__imputer__fill_value | missing | |
|
| preprocessor__cat__imputer__keep_empty_features | False | |
|
| preprocessor__cat__imputer__missing_values | nan | |
|
| preprocessor__cat__imputer__strategy | constant | |
|
| preprocessor__cat__onehot__categories | auto | |
|
| preprocessor__cat__onehot__drop | | |
|
| preprocessor__cat__onehot__dtype | <class 'numpy.float64'> | |
|
| preprocessor__cat__onehot__feature_name_combiner | concat | |
|
| preprocessor__cat__onehot__handle_unknown | ignore | |
|
| preprocessor__cat__onehot__max_categories | | |
|
| preprocessor__cat__onehot__min_frequency | | |
|
| preprocessor__cat__onehot__sparse | deprecated | |
|
| preprocessor__cat__onehot__sparse_output | True | |
|
| classifier__learning_rate | 0.1 | |
|
| classifier__silent | True | |
|
| classifier__max_depth | 4 | |
|
| classifier__n_estimators | 200 | |
|
|
|
</details> |
|
|
|
### Model Plot |
|
|
|
<style>#sk-container-id-1 {color: black;}#sk-container-id-1 pre{padding: 0;}#sk-container-id-1 div.sk-toggleable {background-color: white;}#sk-container-id-1 label.sk-toggleable__label {cursor: pointer;display: block;width: 100%;margin-bottom: 0;padding: 0.3em;box-sizing: border-box;text-align: center;}#sk-container-id-1 label.sk-toggleable__label-arrow:before {content: "▸";float: left;margin-right: 0.25em;color: #696969;}#sk-container-id-1 label.sk-toggleable__label-arrow:hover:before {color: black;}#sk-container-id-1 div.sk-estimator:hover label.sk-toggleable__label-arrow:before {color: black;}#sk-container-id-1 div.sk-toggleable__content {max-height: 0;max-width: 0;overflow: hidden;text-align: left;background-color: #f0f8ff;}#sk-container-id-1 div.sk-toggleable__content pre {margin: 0.2em;color: black;border-radius: 0.25em;background-color: #f0f8ff;}#sk-container-id-1 input.sk-toggleable__control:checked~div.sk-toggleable__content {max-height: 200px;max-width: 100%;overflow: auto;}#sk-container-id-1 input.sk-toggleable__control:checked~label.sk-toggleable__label-arrow:before {content: "▾";}#sk-container-id-1 div.sk-estimator input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 div.sk-label input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 input.sk-hidden--visually {border: 0;clip: rect(1px 1px 1px 1px);clip: rect(1px, 1px, 1px, 1px);height: 1px;margin: -1px;overflow: hidden;padding: 0;position: absolute;width: 1px;}#sk-container-id-1 div.sk-estimator {font-family: monospace;background-color: #f0f8ff;border: 1px dotted black;border-radius: 0.25em;box-sizing: border-box;margin-bottom: 0.5em;}#sk-container-id-1 div.sk-estimator:hover {background-color: #d4ebff;}#sk-container-id-1 div.sk-parallel-item::after {content: "";width: 100%;border-bottom: 1px solid gray;flex-grow: 1;}#sk-container-id-1 div.sk-label:hover label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 div.sk-serial::before {content: "";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: 0;}#sk-container-id-1 div.sk-serial {display: flex;flex-direction: column;align-items: center;background-color: white;padding-right: 0.2em;padding-left: 0.2em;position: relative;}#sk-container-id-1 div.sk-item {position: relative;z-index: 1;}#sk-container-id-1 div.sk-parallel {display: flex;align-items: stretch;justify-content: center;background-color: white;position: relative;}#sk-container-id-1 div.sk-item::before, #sk-container-id-1 div.sk-parallel-item::before {content: "";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: -1;}#sk-container-id-1 div.sk-parallel-item {display: flex;flex-direction: column;z-index: 1;position: relative;background-color: white;}#sk-container-id-1 div.sk-parallel-item:first-child::after {align-self: flex-end;width: 50%;}#sk-container-id-1 div.sk-parallel-item:last-child::after {align-self: flex-start;width: 50%;}#sk-container-id-1 div.sk-parallel-item:only-child::after {width: 0;}#sk-container-id-1 div.sk-dashed-wrapped {border: 1px dashed gray;margin: 0 0.4em 0.5em 0.4em;box-sizing: border-box;padding-bottom: 0.4em;background-color: white;}#sk-container-id-1 div.sk-label label {font-family: monospace;font-weight: bold;display: inline-block;line-height: 1.2em;}#sk-container-id-1 div.sk-label-container {text-align: center;}#sk-container-id-1 div.sk-container {/* jupyter's `normalize.less` sets `[hidden] { display: none; }` but bootstrap.min.css set `[hidden] { display: none !important; }` so we also need the `!important` here to be able to override the default hidden behavior on the sphinx rendered scikit-learn.org. See: https://github.com/scikit-learn/scikit-learn/issues/21755 */display: inline-block !important;position: relative;}#sk-container-id-1 div.sk-text-repr-fallback {display: none;}</style><div id="sk-container-id-1" class="sk-top-container" style="overflow: auto;"><div class="sk-text-repr-fallback"><pre>Pipeline(steps=[('preprocessor',ColumnTransformer(remainder='passthrough',transformers=[('num',Pipeline(steps=[('imputer',SimpleImputer(strategy='median')),('scaler',StandardScaler())]),['rectal_temp', 'pulse','respiratory_rate','nasogastric_reflux_ph','packed_cell_volume','total_protein','abdomo_protein', 'lesion_1','lesion_2', 'lesion_3']),('cat',Pi...OneHotEncoder(handle_unknown='ignore'))]),['surgery', 'age','temp_of_extremities','peripheral_pulse','mucous_membrane','capillary_refill_time','pain', 'peristalsis','abdominal_distention','nasogastric_tube','nasogastric_reflux','rectal_exam_feces','abdomen','abdomo_appearance','surgical_lesion','cp_data'])])),('classifier',<catboost.core.CatBoostClassifier object at 0x000001C4CE4ABF10>)])</pre><b>In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook. <br />On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.</b></div><div class="sk-container" hidden><div class="sk-item sk-dashed-wrapped"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-1" type="checkbox" ><label for="sk-estimator-id-1" class="sk-toggleable__label sk-toggleable__label-arrow">Pipeline</label><div class="sk-toggleable__content"><pre>Pipeline(steps=[('preprocessor',ColumnTransformer(remainder='passthrough',transformers=[('num',Pipeline(steps=[('imputer',SimpleImputer(strategy='median')),('scaler',StandardScaler())]),['rectal_temp', 'pulse','respiratory_rate','nasogastric_reflux_ph','packed_cell_volume','total_protein','abdomo_protein', 'lesion_1','lesion_2', 'lesion_3']),('cat',Pi...OneHotEncoder(handle_unknown='ignore'))]),['surgery', 'age','temp_of_extremities','peripheral_pulse','mucous_membrane','capillary_refill_time','pain', 'peristalsis','abdominal_distention','nasogastric_tube','nasogastric_reflux','rectal_exam_feces','abdomen','abdomo_appearance','surgical_lesion','cp_data'])])),('classifier',<catboost.core.CatBoostClassifier object at 0x000001C4CE4ABF10>)])</pre></div></div></div><div class="sk-serial"><div class="sk-item sk-dashed-wrapped"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-2" type="checkbox" ><label for="sk-estimator-id-2" class="sk-toggleable__label sk-toggleable__label-arrow">preprocessor: ColumnTransformer</label><div class="sk-toggleable__content"><pre>ColumnTransformer(remainder='passthrough',transformers=[('num',Pipeline(steps=[('imputer',SimpleImputer(strategy='median')),('scaler', StandardScaler())]),['rectal_temp', 'pulse', 'respiratory_rate','nasogastric_reflux_ph', 'packed_cell_volume','total_protein', 'abdomo_protein', 'lesion_1','lesion_2', 'lesion_3']),('cat',Pipeline(steps=[('imputer',SimpleI...='missing',strategy='constant')),('onehot',OneHotEncoder(handle_unknown='ignore'))]),['surgery', 'age', 'temp_of_extremities','peripheral_pulse', 'mucous_membrane','capillary_refill_time', 'pain','peristalsis', 'abdominal_distention','nasogastric_tube', 'nasogastric_reflux','rectal_exam_feces', 'abdomen','abdomo_appearance', 'surgical_lesion','cp_data'])])</pre></div></div></div><div class="sk-parallel"><div class="sk-parallel-item"><div class="sk-item"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-3" type="checkbox" ><label for="sk-estimator-id-3" class="sk-toggleable__label sk-toggleable__label-arrow">num</label><div class="sk-toggleable__content"><pre>['rectal_temp', 'pulse', 'respiratory_rate', 'nasogastric_reflux_ph', 'packed_cell_volume', 'total_protein', 'abdomo_protein', 'lesion_1', 'lesion_2', 'lesion_3']</pre></div></div></div><div class="sk-serial"><div class="sk-item"><div class="sk-serial"><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-4" type="checkbox" ><label for="sk-estimator-id-4" class="sk-toggleable__label sk-toggleable__label-arrow">SimpleImputer</label><div class="sk-toggleable__content"><pre>SimpleImputer(strategy='median')</pre></div></div></div><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-5" type="checkbox" ><label for="sk-estimator-id-5" class="sk-toggleable__label sk-toggleable__label-arrow">StandardScaler</label><div class="sk-toggleable__content"><pre>StandardScaler()</pre></div></div></div></div></div></div></div></div><div class="sk-parallel-item"><div class="sk-item"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-6" type="checkbox" ><label for="sk-estimator-id-6" class="sk-toggleable__label sk-toggleable__label-arrow">cat</label><div class="sk-toggleable__content"><pre>['surgery', 'age', 'temp_of_extremities', 'peripheral_pulse', 'mucous_membrane', 'capillary_refill_time', 'pain', 'peristalsis', 'abdominal_distention', 'nasogastric_tube', 'nasogastric_reflux', 'rectal_exam_feces', 'abdomen', 'abdomo_appearance', 'surgical_lesion', 'cp_data']</pre></div></div></div><div class="sk-serial"><div class="sk-item"><div class="sk-serial"><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-7" type="checkbox" ><label for="sk-estimator-id-7" class="sk-toggleable__label sk-toggleable__label-arrow">SimpleImputer</label><div class="sk-toggleable__content"><pre>SimpleImputer(fill_value='missing', strategy='constant')</pre></div></div></div><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-8" type="checkbox" ><label for="sk-estimator-id-8" class="sk-toggleable__label sk-toggleable__label-arrow">OneHotEncoder</label><div class="sk-toggleable__content"><pre>OneHotEncoder(handle_unknown='ignore')</pre></div></div></div></div></div></div></div></div><div class="sk-parallel-item"><div class="sk-item"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-9" type="checkbox" ><label for="sk-estimator-id-9" class="sk-toggleable__label sk-toggleable__label-arrow">remainder</label><div class="sk-toggleable__content"><pre>[]</pre></div></div></div><div class="sk-serial"><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-10" type="checkbox" ><label for="sk-estimator-id-10" class="sk-toggleable__label sk-toggleable__label-arrow">passthrough</label><div class="sk-toggleable__content"><pre>passthrough</pre></div></div></div></div></div></div></div></div><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable__control sk-hidden--visually" id="sk-estimator-id-11" type="checkbox" ><label for="sk-estimator-id-11" class="sk-toggleable__label sk-toggleable__label-arrow">CatBoostClassifier</label><div class="sk-toggleable__content"><pre><catboost.core.CatBoostClassifier object at 0x000001C4CE4ABF10></pre></div></div></div></div></div></div></div> |
|
|
|
## Evaluation Results |
|
|
|
| Metric | Value | |
|
|----------|----------| |
|
| accuracy | 0.744939 | |
|
| f1 score | 0.744939 | |
|
|
|
### Confusion Matrix |
|
|
|
![Confusion Matrix](confusion_matrix.png) |
|
|
|
# How to Get Started with the Model |
|
|
|
[More Information Needed] |
|
|
|
# Model Card Authors |
|
|
|
kmposkid |
|
|
|
# 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] |
|
``` |
|
|