joshuasundance
commited on
Commit
•
1e7a595
1
Parent(s):
a4331d2
Update README.md
Browse files
README.md
CHANGED
@@ -68,8 +68,8 @@ This model was trained within the context of a larger system for ABSA, which loo
|
|
68 |
- **Sentence Transformer body:** [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2)
|
69 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
70 |
- **spaCy Model:** en_core_web_sm
|
71 |
-
- **SetFitABSA Aspect Model:** [
|
72 |
-
- **SetFitABSA Polarity Model:** [
|
73 |
- **Maximum Sequence Length:** 384 tokens
|
74 |
- **Number of Classes:** 4 classes
|
75 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
@@ -114,8 +114,8 @@ from setfit import AbsaModel
|
|
114 |
|
115 |
# Download from the 🤗 Hub
|
116 |
model = AbsaModel.from_pretrained(
|
117 |
-
"
|
118 |
-
"
|
119 |
)
|
120 |
# Run inference
|
121 |
preds = model("The food was great, but the venue is just way too busy.")
|
|
|
68 |
- **Sentence Transformer body:** [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2)
|
69 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
70 |
- **spaCy Model:** en_core_web_sm
|
71 |
+
- **SetFitABSA Aspect Model:** [joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-aspect](https://huggingface.co/joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-aspect)
|
72 |
+
- **SetFitABSA Polarity Model:** [joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-polarity](https://huggingface.co/joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-polarity)
|
73 |
- **Maximum Sequence Length:** 384 tokens
|
74 |
- **Number of Classes:** 4 classes
|
75 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
|
|
114 |
|
115 |
# Download from the 🤗 Hub
|
116 |
model = AbsaModel.from_pretrained(
|
117 |
+
"joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-aspect",
|
118 |
+
"joshuasundance/setfit-absa-all-mpnet-base-v2-laptops-polarity",
|
119 |
)
|
120 |
# Run inference
|
121 |
preds = model("The food was great, but the venue is just way too busy.")
|