system HF staff commited on
Commit
796faec
·
1 Parent(s): 8d0ee14

Commit From AutoTrain

Browse files
Files changed (4) hide show
  1. .gitattributes +3 -0
  2. README.md +51 -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,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - tabular
5
+ - classification
6
+ - tabular-classification
7
+ datasets:
8
+ - dozata/autotrain-data-petfinder-demo
9
+ co2_eq_emissions:
10
+ emissions: 0.49257909869639516
11
+ ---
12
+
13
+ # Model Trained Using AutoTrain
14
+
15
+ - Problem type: Multi-class Classification
16
+ - Model ID: 87059143318
17
+ - CO2 Emissions (in grams): 0.4926
18
+
19
+ ## Validation Metrics
20
+
21
+ - Loss: 4.209
22
+ - Accuracy: 0.033
23
+ - Macro F1: 0.003
24
+ - Micro F1: 0.033
25
+ - Weighted F1: 0.010
26
+ - Macro Precision: 0.003
27
+ - Micro Precision: 0.033
28
+ - Weighted Precision: 0.008
29
+ - Macro Recall: 0.011
30
+ - Micro Recall: 0.033
31
+ - Weighted Recall: 0.033
32
+
33
+ ## Usage
34
+
35
+ ```python
36
+ import json
37
+ import joblib
38
+ import pandas as pd
39
+
40
+ model = joblib.load('model.joblib')
41
+ config = json.load(open('config.json'))
42
+
43
+ features = config['features']
44
+
45
+ # data = pd.read_csv("data.csv")
46
+ data = data[features]
47
+ data.columns = ["feat_" + str(col) for col in data.columns]
48
+
49
+ predictions = model.predict(data) # or model.predict_proba(data)
50
+
51
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"features": ["Subject Focus", "Eyes", "Face", "Near", "Action", "Accessory", "Group", "Collage", "Human", "Occlusion", "Info", "Blur"], "targets": ["target"], "model_type": "xgboost", "target_mapping": {"1": 0, "10": 1, "100": 2, "11": 3, "12": 4, "13": 5, "14": 6, "15": 7, "16": 8, "17": 9, "18": 10, "19": 11, "2": 12, "20": 13, "21": 14, "22": 15, "23": 16, "24": 17, "25": 18, "26": 19, "27": 20, "28": 21, "29": 22, "3": 23, "30": 24, "31": 25, "32": 26, "33": 27, "34": 28, "35": 29, "36": 30, "37": 31, "38": 32, "39": 33, "4": 34, "40": 35, "41": 36, "42": 37, "43": 38, "44": 39, "45": 40, "46": 41, "47": 42, "48": 43, "49": 44, "5": 45, "50": 46, "51": 47, "52": 48, "53": 49, "54": 50, "55": 51, "56": 52, "57": 53, "58": 54, "59": 55, "6": 56, "60": 57, "61": 58, "62": 59, "63": 60, "64": 61, "65": 62, "66": 63, "67": 64, "68": 65, "69": 66, "7": 67, "70": 68, "71": 69, "72": 70, "73": 71, "74": 72, "75": 73, "76": 74, "77": 75, "78": 76, "79": 77, "8": 78, "80": 79, "81": 80, "82": 81, "83": 82, "84": 83, "85": 84, "86": 85, "87": 86, "88": 87, "89": 88, "9": 89, "90": 90, "91": 91, "92": 92, "93": 93, "94": 94, "95": 95, "96": 96, "97": 97, "98": 98, "99": 99}}
model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2bcd7fa9dae8228d8ed28f76f6fbda2f01afeaaea4df7acbf184f97bcc7ba6b6
3
+ size 32471189