system HF staff commited on
Commit
3055a9b
1 Parent(s): 9c3c03b

Commit From AutoTrain

Browse files
Files changed (4) hide show
  1. .gitattributes +3 -0
  2. README.md +45 -0
  3. config.json +1 -0
  4. model.joblib +3 -0
.gitattributes CHANGED
@@ -25,3 +25,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ *.bin.* 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,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - tabular
5
+ - classification
6
+ - tabular-classification
7
+ datasets:
8
+ - abhishek/autotrain-data-adult-train
9
+ co2_eq_emissions: 0.12693590577861977
10
+ ---
11
+
12
+ # Model Trained Using AutoTrain
13
+
14
+ - Problem type: Binary Classification
15
+ - Model ID: 9725286
16
+ - CO2 Emissions (in grams): 0.12693590577861977
17
+
18
+ ## Validation Metrics
19
+
20
+ - Loss: 0.26716182056213406
21
+ - Accuracy: 0.8750191923844618
22
+ - Precision: 0.7840481565086531
23
+ - Recall: 0.6641172721478649
24
+ - AUC: 0.9345322809861784
25
+ - F1: 0.7191166321601105
26
+
27
+ ## Usage
28
+
29
+ ```python
30
+ import json
31
+ import joblib
32
+ import pandas as pd
33
+
34
+ model = joblib.load('model.joblib')
35
+ config = json.load(open('config.json'))
36
+
37
+ features = config['features']
38
+
39
+ # data = pd.read_csv("data.csv")
40
+ data = data[features]
41
+ data.columns = ["feat_" + str(col) for col in data.columns]
42
+
43
+ predictions = model.predict(data) # or model.predict_proba(data)
44
+
45
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
1
+ {"features": ["age", "fnlwgt", "education.num", "capital.gain", "capital.loss", "hours.per.week", "workclass", "education", "marital.status", "occupation", "relationship", "race", "sex", "native.country"], "targets": ["target"], "model_type": "xgboost", "target_mapping": {"<=50K": 0, ">50K": 1}}
model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:174d326fff8b2b8c21d7c3116e9c463849779dca66f38a841a23b1b0e549c506
3
+ size 9978252