system HF staff commited on
Commit
556407e
1 Parent(s): b0764c4

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
+ - rajistics/autotrain-data-Adult
9
+ co2_eq_emissions: 38.42484725553464
10
+ ---
11
+
12
+ # Model Trained Using AutoTrain
13
+
14
+ - Problem type: Binary Classification
15
+ - Model ID: 934630783
16
+ - CO2 Emissions (in grams): 38.42484725553464
17
+
18
+ ## Validation Metrics
19
+
20
+ - Loss: 0.2984429822985684
21
+ - Accuracy: 0.8628221244500315
22
+ - Precision: 0.7873263888888888
23
+ - Recall: 0.5908794788273616
24
+ - AUC: 0.9182195921357326
25
+ - F1: 0.6751023446222553
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": "extra_trees", "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:77cbb95ff6cedad10be5946d2899f97c6ad77a656f01aae88aea57270f5d6863
3
+ size 638485433