Update README.md
Browse files
README.md
CHANGED
@@ -62,8 +62,8 @@ This model was trained within the context of a larger system for ABSA, which loo
|
|
62 |
- **Sentence Transformer body:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)
|
63 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
64 |
- **spaCy Model:** en_core_web_sm
|
65 |
-
- **SetFitABSA Aspect Model:** [
|
66 |
-
- **SetFitABSA Polarity Model:** [
|
67 |
- **Maximum Sequence Length:** 256 tokens
|
68 |
- **Number of Classes:** 2 classes
|
69 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
@@ -106,8 +106,8 @@ from setfit import AbsaModel
|
|
106 |
|
107 |
# Download from the 🤗 Hub
|
108 |
model = AbsaModel.from_pretrained(
|
109 |
-
"
|
110 |
-
"
|
111 |
)
|
112 |
# Run inference
|
113 |
preds = model("The food was great, but the venue is just way too busy.")
|
|
|
62 |
- **Sentence Transformer body:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)
|
63 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
64 |
- **spaCy Model:** en_core_web_sm
|
65 |
+
- **SetFitABSA Aspect Model:** [joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-aspect](https://huggingface.co/joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-aspect)
|
66 |
+
- **SetFitABSA Polarity Model:** [joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-polarity](https://huggingface.co/joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-polarity)
|
67 |
- **Maximum Sequence Length:** 256 tokens
|
68 |
- **Number of Classes:** 2 classes
|
69 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
|
|
106 |
|
107 |
# Download from the 🤗 Hub
|
108 |
model = AbsaModel.from_pretrained(
|
109 |
+
"joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-aspect",
|
110 |
+
"joshuasundance/setfit-absa-all-MiniLM-L6-v2-laptops-polarity",
|
111 |
)
|
112 |
# Run inference
|
113 |
preds = model("The food was great, but the venue is just way too busy.")
|