system HF staff commited on
Commit
c4bbdcd
1 Parent(s): 9d0f044

Commit From AutoTrain

Browse files
Files changed (4) hide show
  1. .gitattributes +3 -0
  2. README.md +46 -0
  3. config.json +1 -0
  4. model.joblib +3 -0
.gitattributes CHANGED
@@ -32,3 +32,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ *.bin.* filter=lfs diff=lfs merge=lfs -text
36
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
37
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - tabular
5
+ - classification
6
+ - tabular-classification
7
+ datasets:
8
+ - harithapliyal/autotrain-data-tatanic-survival
9
+ co2_eq_emissions:
10
+ emissions: 0.004107493848653723
11
+ ---
12
+
13
+ # Model Trained Using AutoTrain
14
+
15
+ - Problem type: Binary Classification
16
+ - Model ID: 51030121311
17
+ - CO2 Emissions (in grams): 0.0041
18
+
19
+ ## Validation Metrics
20
+
21
+ - Loss: 0.358
22
+ - Accuracy: 0.872
23
+ - Precision: 0.859
24
+ - Recall: 0.797
25
+ - AUC: 0.903
26
+ - F1: 0.827
27
+
28
+ ## Usage
29
+
30
+ ```python
31
+ import json
32
+ import joblib
33
+ import pandas as pd
34
+
35
+ model = joblib.load('model.joblib')
36
+ config = json.load(open('config.json'))
37
+
38
+ features = config['features']
39
+
40
+ # data = pd.read_csv("data.csv")
41
+ data = data[features]
42
+ data.columns = ["feat_" + str(col) for col in data.columns]
43
+
44
+ predictions = model.predict(data) # or model.predict_proba(data)
45
+
46
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"features": ["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:a72f9cc37bedb8f3f61067b151a9e39160511246af55468c8601a1a7289e7669
3
+ size 735042