system HF staff commited on
Commit
7790cf4
1 Parent(s): 37bb7bb

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
+ - marac5/autotrain-data-builty-2-table-searcher
9
+ co2_eq_emissions:
10
+ emissions: 7.738626953271278
11
+ ---
12
+
13
+ # Model Trained Using AutoTrain
14
+
15
+ - Problem type: Binary Classification
16
+ - Model ID: 3373492718
17
+ - CO2 Emissions (in grams): 7.7386
18
+
19
+ ## Validation Metrics
20
+
21
+ - Loss: 0.360
22
+ - Accuracy: 0.871
23
+ - Precision: 0.871
24
+ - Recall: 0.783
25
+ - AUC: 0.916
26
+ - F1: 0.824
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", "Siblings/Spouses Aboard", "Parents/Children Aboard", "Fare", "Sex"], "targets": ["target"], "model_type": "random_forest", "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:9a4cd4bcafef37307d7277f5e33a757aefc8fa110b45e08091b498829e2ed410
3
+ size 28708498