system HF staff commited on
Commit
9fdbf61
1 Parent(s): 9de518a

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
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ *.bin.* filter=lfs diff=lfs merge=lfs -text
37
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
38
+ 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
+ - jrosenzw/autotrain-data-diabetes-detection-2
9
+ co2_eq_emissions:
10
+ emissions: 0.6960832080920549
11
+ ---
12
+
13
+ # Model Trained Using AutoTrain
14
+
15
+ - Problem type: Binary Classification
16
+ - Model ID: 74371139581
17
+ - CO2 Emissions (in grams): 0.6961
18
+
19
+ ## Validation Metrics
20
+
21
+ - Loss: 0.468
22
+ - Accuracy: 0.753
23
+ - Precision: 0.667
24
+ - Recall: 0.593
25
+ - AUC: 0.839
26
+ - F1: 0.627
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": ["Pregnancies", "Glucose", "BloodPressure", "SkinThickness", "Insulin", "BMI", "DiabetesPedigreeFunction", "Age"], "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:1a43efd0690e94126c7bc5cee88a163eece5c5c1b063a03b6c7c9b4a37ec34da
3
+ size 307983