victor HF Staff commited on
Commit
e7b4c54
·
verified ·
1 Parent(s): 755d21e

Add wine type classifier

Browse files
Files changed (3) hide show
  1. feature_importances.json +14 -0
  2. metadata.json +31 -0
  3. model.pkl +3 -0
feature_importances.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fixed_acidity": 0.011216096291593433,
3
+ "volatile_acidity": 0.022733954817831945,
4
+ "citric_acid": 0.002625650838238,
5
+ "residual_sugar": 0.0036831021517609356,
6
+ "chlorides": 0.3125462863195496,
7
+ "free_sulfur_dioxide": 0.002256796690168855,
8
+ "total_sulfur_dioxide": 0.5806053706806034,
9
+ "density": 0.033983885290608154,
10
+ "pH": 0.012845647512315176,
11
+ "sulphates": 0.013827989107109365,
12
+ "alcohol": 0.0033643561589295143,
13
+ "quality": 0.00031086414129171785
14
+ }
metadata.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "GradientBoostingClassifier",
3
+ "sklearn_version": "1.8.0",
4
+ "features": [
5
+ "fixed_acidity",
6
+ "volatile_acidity",
7
+ "citric_acid",
8
+ "residual_sugar",
9
+ "chlorides",
10
+ "free_sulfur_dioxide",
11
+ "total_sulfur_dioxide",
12
+ "density",
13
+ "pH",
14
+ "sulphates",
15
+ "alcohol",
16
+ "quality"
17
+ ],
18
+ "target": "is_red",
19
+ "labels": {
20
+ "0": "White Wine",
21
+ "1": "Red Wine"
22
+ },
23
+ "metrics": {
24
+ "test_accuracy": 0.9923,
25
+ "test_f1": 0.9949,
26
+ "train_accuracy": 1.0
27
+ },
28
+ "dataset": "mstz/wine",
29
+ "n_train": 5197,
30
+ "n_test": 1300
31
+ }
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76a6fb618e4eb4a60d1a3f19bbabeacaa71df27df6c78d446913632c0eaa5e17
3
+ size 401141