diff --git a/fasttext_amazon_reviews/README.md b/fasttext_amazon_reviews/README.md index e2426de..d0bc8e5 100644 --- a/fasttext_amazon_reviews/README.md +++ b/fasttext_amazon_reviews/README.md @@ -1,10 +1,6 @@ # Training a Fasttext model on the amazon reviews dataset -This example demonstrates the use of the following module below from cleanlab: - -- [cleanlab.models.fasttext.py](https://github.com/cleanlab/cleanlab/blob/master/cleanlab/models/fasttext.py) - -The code is adapted from cleanlab v1 examples (see `contrib/v1` folder). +This example demonstrates how to wrap a Fasttext model to be sklearn-compatible for direct compatibility with `cleanlab.classification`. The wrapper code is in **fasttext_wrapper.py** and the notebook demonstrates how to apply it to a text classification dataset. ## Instructions @@ -20,4 +16,4 @@ Run bash script below to download all the data. $ bash ./download_data.sh ``` -Start Jupyter Lab and run the notebook: `fasttext_amazon_reviews.ipynb` +Start Jupyter and run the notebook: `fasttext_amazon_reviews.ipynb`