wine-type-classifier / metadata.json
victor's picture
victor HF Staff
Add metadata and feature importances
4c1be3d verified
{
"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
}