diff --git a/README.md b/README.md index 588c363..b982228 100644 --- a/README.md +++ b/README.md @@ -122,7 +122,7 @@ Alternatively, you can `rsdk push ` via CLI, given that you h ### Adapters -#### 🤗 Hugging Face +#### Hugging Face 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: ```python @@ -179,7 +179,7 @@ trainer.save_model("path/to/model") 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: ```python -from refinery.adapter.embedders import build_classification_dataset +from refinery.adapter.sklearn import build_classification_dataset from sklearn.tree import DecisionTreeClassifier data = build_classification_dataset(client, "headline", "__clickbait", "distilbert-base-uncased")