Update README.md
Browse files
README.md
CHANGED
@@ -6,28 +6,4 @@ datasets:
|
|
6 |
metrics:
|
7 |
- accuracy
|
8 |
pipeline_tag: tabular-classification
|
9 |
-
---
|
10 |
-
|
11 |
-
# Model Trained Using AutoTrain
|
12 |
-
|
13 |
-
- Problem type: Multi-class Classification
|
14 |
-
|
15 |
-
## Usage
|
16 |
-
|
17 |
-
```python
|
18 |
-
import json
|
19 |
-
import joblib
|
20 |
-
import pandas as pd
|
21 |
-
|
22 |
-
model = joblib.load('iris_svm.joblib')
|
23 |
-
config = json.load(open('config.json'))
|
24 |
-
|
25 |
-
features = config['features']
|
26 |
-
|
27 |
-
# data = pd.read_csv("data.csv")
|
28 |
-
data = data[features]
|
29 |
-
data.columns = ["feat_" + str(col) for col in data.columns]
|
30 |
-
|
31 |
-
predictions = model.predict(data) # or model.predict_proba(data)
|
32 |
-
|
33 |
-
```
|
|
|
6 |
metrics:
|
7 |
- accuracy
|
8 |
pipeline_tag: tabular-classification
|
9 |
+
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|