system HF staff commited on
Commit
d489fb3
1 Parent(s): 6fff597

Commit From AutoTrain

Browse files
Files changed (4) hide show
  1. .gitattributes +2 -0
  2. README.md +43 -0
  3. config.json +1 -0
  4. model.joblib +3 -0
.gitattributes CHANGED
@@ -26,3 +26,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
26
  *.zip filter=lfs diff=lfs merge=lfs -text
27
  *.zstandard filter=lfs diff=lfs merge=lfs -text
28
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
26
  *.zip filter=lfs diff=lfs merge=lfs -text
27
  *.zstandard filter=lfs diff=lfs merge=lfs -text
28
  *tfevents* filter=lfs diff=lfs merge=lfs -text
29
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
30
+ *.pkl filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - tabular
5
+ - classification
6
+ - structured-data-classification
7
+ datasets:
8
+ - vabadeh213/autotrain-data-titanic
9
+ co2_eq_emissions: 0.00509303545772981
10
+ ---
11
+
12
+ # Model Trained Using AutoTrain
13
+
14
+ - Problem type: Binary Classification
15
+ - Model ID: 744222727
16
+ - CO2 Emissions (in grams): 0.00509303545772981
17
+
18
+ ## Validation Metrics
19
+
20
+ - Loss: 0.40596098709549455
21
+ - Accuracy: 0.8378378378378378
22
+ - Precision: 0.8518518518518519
23
+ - Recall: 0.92
24
+ - AUC: 0.8866666666666667
25
+ - F1: 0.8846153846153846
26
+
27
+ ## Usage
28
+
29
+ ```python
30
+ import json
31
+ import joblib
32
+
33
+ model = joblib.load('model.joblib')
34
+ config = json.load(open('config.json'))
35
+
36
+ features = config['features']
37
+
38
+ # data = pd.read_csv("data.csv")
39
+ data = data[features]
40
+
41
+ predictions = model.predict(data) # or model.predict_proba(data)
42
+
43
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
 
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 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c9a3c7c1e0221c10ddf70a1bb70658b6e906020924630ccd063a6a0b3063105
3
+ size 298553