Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model/stroke_risk_prediction/model.cb filter=lfs diff=lfs merge=lfs -text

data/metadata.json

@@ -0,0 +1,84 @@
+{
+    "random_seed": 14,
+    "test_size": 0.2,
+    "target_feature": "stroke",
+    "original_features": [
+        "gender",
+        "age",
+        "hypertension",
+        "heart_disease",
+        "ever_married",
+        "work_type",
+        "Residence_type",
+        "avg_glucose_level",
+        "bmi",
+        "smoking_status"
+    ],
+    "binary_encoding": {
+        "columns": [
+            "ever_married"
+        ],
+        "mapping": {
+            "No": 0,
+            "Yes": 1
+        }
+    },
+    "imputation": {
+        "method": "SimpleImputer",
+        "strategy": "median",
+        "columns": [
+            "bmi"
+        ]
+    },
+    "scaling": {
+        "method": "StandardScaler",
+        "columns": [
+            "avg_glucose_level",
+            "bmi"
+        ]
+    },
+    "encoding": {
+        "method": "OneHotEncoder",
+        "columns": [
+            "gender",
+            "work_type",
+            "residence_type",
+            "smoking_status"
+        ],
+        "verbose_feature_names": false
+    },
+    "processed_features": [
+        "stroke",
+        "ever_married",
+        "heart_disease",
+        "hypertension",
+        "age",
+        "smoking_status_smokes",
+        "smoking_status_never_smoked",
+        "smoking_status_formerly_smoked",
+        "smoking_status_unknown",
+        "residence_type_urban",
+        "residence_type_rural",
+        "work_type_children",
+        "work_type_self-employed",
+        "work_type_private",
+        "work_type_never_worked",
+        "work_type_govt_job",
+        "gender_other",
+        "gender_male",
+        "gender_female",
+        "bmi",
+        "avg_glucose_level"
+    ],
+    "artifacts_dir": "data/artifacts",
+    "processed_data_dir": "data/processed",
+    "train_shape": [
+        4088,
+        21
+    ],
+    "test_shape": [
+        1022,
+        21
+    ],
+    "timestamp": "2026-05-16T16:59:22.957524"
+}

model/stroke_risk_prediction/MLmodel

@@ -0,0 +1,48 @@
+artifact_path: /media/vini/DATA/Git/ml_full_project/mlruns/3/models/m-39de93ac07234ce09b688ba8df87c48d/artifacts
+flavors:
+  catboost:
+    catboost_version: 1.2.10
+    code: null
+    data: model.cb
+    model_type: CatBoostClassifier
+    save_format: cbm
+  python_function:
+    data: model.cb
+    env:
+      conda: conda.yaml
+      virtualenv: python_env.yaml
+    loader_module: mlflow.catboost
+    python_version: 3.12.11
+is_signature_from_type_hint: false
+mlflow_version: 3.10.1
+model_id: m-39de93ac07234ce09b688ba8df87c48d
+model_size_bytes: 2012951
+model_uuid: m-39de93ac07234ce09b688ba8df87c48d
+prompts: null
+run_id: 5c4cf23fc7c94efa8b9ffcbf2265c7a1
+saved_input_example_info:
+  artifact_path: input_example.json
+  pandas_orient: split
+  serving_input_path: serving_input_example.json
+  type: dataframe
+signature:
+  inputs: '[{"type": "double", "name": "avg_glucose_level", "required": true}, {"type":
+    "double", "name": "bmi", "required": true}, {"type": "double", "name": "gender_female",
+    "required": true}, {"type": "double", "name": "gender_male", "required": true},
+    {"type": "double", "name": "gender_other", "required": true}, {"type": "double",
+    "name": "work_type_govt_job", "required": true}, {"type": "double", "name": "work_type_never_worked",
+    "required": true}, {"type": "double", "name": "work_type_private", "required":
+    true}, {"type": "double", "name": "work_type_self-employed", "required": true},
+    {"type": "double", "name": "work_type_children", "required": true}, {"type": "double",
+    "name": "residence_type_rural", "required": true}, {"type": "double", "name":
+    "residence_type_urban", "required": true}, {"type": "double", "name": "smoking_status_unknown",
+    "required": true}, {"type": "double", "name": "smoking_status_formerly_smoked",
+    "required": true}, {"type": "double", "name": "smoking_status_never_smoked", "required":
+    true}, {"type": "double", "name": "smoking_status_smokes", "required": true},
+    {"type": "double", "name": "age", "required": true}, {"type": "double", "name":
+    "hypertension", "required": true}, {"type": "double", "name": "heart_disease",
+    "required": true}, {"type": "double", "name": "ever_married", "required": true}]'
+  outputs: '[{"type": "long", "required": true}]'
+  params: null
+type_hint_from_example: false
+utc_time_created: '2026-05-16 16:06:46.813357'

model/stroke_risk_prediction/conda.yaml

@@ -0,0 +1,14 @@
+channels:
+- conda-forge
+dependencies:
+- python=3.12.11
+- pip<=26.0.1
+- pip:
+  - mlflow==3.10.1
+  - catboost==1.2.10
+  - colorama==0.4.6
+  - ipython==9.12.0
+  - numpy==2.4.4
+  - pandas==2.3.3
+  - scipy==1.17.1
+name: mlflow-env

model/stroke_risk_prediction/model.cb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8f09d323dc3bccd3276d6fe73560f824dec1ffc44eb1112ee0cfbc033b71296f
+size 200392

model/stroke_risk_prediction/python_env.yaml

@@ -0,0 +1,7 @@
+python: 3.12.11
+build_dependencies:
+- pip==26.0.1
+- setuptools==82.0.1
+- wheel
+dependencies:
+- -r requirements.txt

model/stroke_risk_prediction/registered_model_meta

@@ -0,0 +1,2 @@
+model_name: stroke_risk_prediction
+model_version: '1'

model/stroke_risk_prediction/requirements.txt

@@ -0,0 +1,7 @@
+mlflow==3.10.1
+catboost==1.2.10
+colorama==0.4.6
+ipython==9.12.0
+numpy==2.4.4
+pandas==2.3.3
+scipy==1.17.1

preprocessor/preprocessor.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef69e32a290c075d3220709ad81cd7375b8654c30fc6c80f7c72fd1c5f153d3b
+size 4063