Hugging Face's logo

victor
/
wine-type-classifier

Model card Files
xet
 Community
wine-type-classifier
File size: 573 Bytes
e7b4c54
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4c1be3d
 
e7b4c54
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
{
  "model_type": "GradientBoostingClassifier",
  "sklearn_version": "1.8.0",
  "features": [
    "fixed_acidity",
    "volatile_acidity",
    "citric_acid",
    "residual_sugar",
    "chlorides",
    "free_sulfur_dioxide",
    "total_sulfur_dioxide",
    "density",
    "pH",
    "sulphates",
    "alcohol",
    "quality"
  ],
  "target": "is_red",
  "labels": {
    "0": "Red Wine",
    "1": "White Wine"
  },
  "metrics": {
    "test_accuracy": 0.9923,
    "test_f1": 0.9949,
    "train_accuracy": 1.0
  },
  "dataset": "mstz/wine",
  "n_train": 5197,
  "n_test": 1300
}