Skip to content

Commit a68a99e

Browse files
Changed type in the sklearn section of the readme (#19)
* Changed type in the sklearn section of the readme * Update README.md remove emoji from hugging face link Co-authored-by: Johannes Hötter <johannes.hoetter@kern.ai>
1 parent 4ba7adc commit a68a99e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ Alternatively, you can `rsdk push <path-to-your-file>` via CLI, given that you h
122122

123123
### Adapters
124124

125-
#### 🤗 Hugging Face
125+
#### Hugging Face
126126
Transformers are great, but often times, you want to finetune them for your downstream task. With *refinery*, you can do so easily by letting the SDK build the dataset for you that you can use as a plug-and-play base for your training:
127127

128128
```python
@@ -179,7 +179,7 @@ trainer.save_model("path/to/model")
179179
You can use *refinery* to directly pull data into a format you can apply for building [sklearn](https://github.com/scikit-learn/scikit-learn) models. This can look as follows:
180180

181181
```python
182-
from refinery.adapter.embedders import build_classification_dataset
182+
from refinery.adapter.sklearn import build_classification_dataset
183183
from sklearn.tree import DecisionTreeClassifier
184184

185185
data = build_classification_dataset(client, "headline", "__clickbait", "distilbert-base-uncased")

0 commit comments

Comments
 (0)