Commit From AutoTrain

.gitattributes

@@ -26,3 +26,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zstandard filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.tar.gz filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,43 @@
+---
+tags:
+- autotrain
+- tabular
+- classification
+- structured-data-classification
+datasets:
+- vabadeh213/autotrain-data-titanic
+co2_eq_emissions: 0.00509303545772981
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Binary Classification
+- Model ID: 744222727
+- CO2 Emissions (in grams): 0.00509303545772981
+
+## Validation Metrics
+
+- Loss: 0.40596098709549455
+- Accuracy: 0.8378378378378378
+- Precision: 0.8518518518518519
+- Recall: 0.92
+- AUC: 0.8866666666666667
+- F1: 0.8846153846153846
+
+## Usage
+
+```python
+import json
+import joblib
+
+model = joblib.load('model.joblib')
+config = json.load(open('config.json'))
+
+features = config['features']
+
+# data = pd.read_csv("data.csv")
+data = data[features]
+
+predictions = model.predict(data)  # or model.predict_proba(data)
+
+```

config.json

@@ -0,0 +1 @@
+{"features": ["PassengerId", "Pclass", "Age", "SibSp", "Parch", "Fare", "Name", "Sex", "Ticket", "Cabin", "Embarked"], "targets": ["target"], "model_type": "xgboost", "target_mapping": {"0": 0, "1": 1}}

model.joblib

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