Update README.md
Browse files
README.md
CHANGED
@@ -23,5 +23,24 @@ widget:
|
|
23 |
- 0
|
24 |
- 0
|
25 |
- 2
|
26 |
-
|
27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
- 0
|
24 |
- 0
|
25 |
- 2
|
26 |
+
tags:
|
27 |
+
- sklearn
|
28 |
+
---
|
29 |
+
### How to use
|
30 |
+
|
31 |
+
```python
|
32 |
+
from huggingface_hub import hf_hub_url, cached_download
|
33 |
+
import joblib
|
34 |
+
import pandas as pd
|
35 |
+
from sklearn.datasets import load_iris
|
36 |
+
from sklearn.model_selection import train_test_split
|
37 |
+
|
38 |
+
REPO_ID = "d2i-pti-iu/test_svc_model"
|
39 |
+
FILENAME = "iris_svm.joblib"
|
40 |
+
model = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
|
41 |
+
iris = load_iris()
|
42 |
+
|
43 |
+
X = iris.data[:3]
|
44 |
+
# model is a `sklearn.pipeline.Pipeline`
|
45 |
+
labels = model.predict(X)
|
46 |
+
```
|