Commit From AutoTrain

.gitattributes

@@ -32,3 +32,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.tar.gz filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,51 @@
+---
+tags:
+- autotrain
+- tabular
+- classification
+- tabular-classification
+datasets:
+- reesu/autotrain-data-wine_quality
+co2_eq_emissions:
+  emissions: 0.8738507920594603
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Multi-class Classification
+- Model ID: 3195889865
+- CO2 Emissions (in grams): 0.8739
+
+## Validation Metrics
+
+- Loss: 15.722
+- Accuracy: 0.545
+- Macro F1: 0.226
+- Micro F1: 0.545
+- Weighted F1: 0.500
+- Macro Precision: 0.260
+- Micro Precision: 0.545
+- Weighted Precision: 0.507
+- Macro Recall: 0.240
+- Micro Recall: 0.545
+- Weighted Recall: 0.545
+
+## Usage
+
+```python
+import json
+import joblib
+import pandas as pd
+
+model = joblib.load('model.joblib')
+config = json.load(open('config.json'))
+
+features = config['features']
+
+# data = pd.read_csv("data.csv")
+data = data[features]
+data.columns = ["feat_" + str(col) for col in data.columns]
+
+predictions = model.predict(data)  # or model.predict_proba(data)
+
+```

config.json

@@ -0,0 +1 @@
+{"features": ["Id", "fixed acidity", "volatile acidity", "citric acid", "residual sugar", "chlorides", "free sulfur dioxide", "total sulfur dioxide", "density", "pH", "sulphates", "alcohol"], "targets": ["target"], "model_type": "svm", "target_mapping": {"3": 0, "4": 1, "5": 2, "6": 3, "7": 4, "8": 5}}

model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27df316282535696b527ad47ff04bf39ba1c5ccc92a6fda2446999310f200859
+size 4563